NumPy 是 Python 中一个强大的数值计算库,它提供了多维数组对象以及一系列的数学函数。以下是一些基本的教程内容,帮助您开始使用 NumPy。

安装NumPy

在开始之前,请确保您已经安装了 NumPy。您可以使用以下命令进行安装:

pip install numpy

基本操作

创建数组

NumPy 提供了多种创建数组的方法。以下是一个简单的例子:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr)

数组索引

与 Python 列表类似,NumPy 数组也可以通过索引来访问元素。

print(arr[0])  # 输出: 1

数组切片

您可以使用切片来获取数组的一部分。

print(arr[1:4])  # 输出: [2 3 4]

数组操作

NumPy 提供了丰富的数组操作功能,如加法、减法、乘法和除法。

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

print(arr1 + arr2)  # 输出: [ 5  7  9]

高级功能

NumPy 支持高级功能,如线性代数、傅里叶变换等。

线性代数

import numpy as np

A = np.array([[1, 2], [3, 4]])
B = np.array([[2, 0], [1, 2]])

print(np.dot(A, B))  # 输出: [[4 4]
                       #         [10 8]]

傅里叶变换

import numpy as np

x = np.linspace(0, 2 * np.pi, 100)
y = np.sin(x)

print(np.fft.fft(y))  # 输出: fft变换后的结果

扩展阅读

如果您想了解更多关于 NumPy 的信息,请访问我们的 NumPy 教程

NumPy Logo