NumPy 是 Python 中用于科学计算的库,它提供了强大的多维数组对象和一系列用于处理数组的函数。
安装 NumPy
首先,确保你已经安装了 Python。然后,可以通过以下命令安装 NumPy:
pip install numpy
基本使用
创建数组
import numpy as np
# 创建一个一维数组
array_1d = np.array([1, 2, 3, 4, 5])
# 创建一个二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
数组操作
NumPy 提供了丰富的数组操作函数,如:
array.shape
:获取数组的形状array.size
:获取数组的元素数量array.sum()
:计算数组的所有元素之和
print("Shape:", array_1d.shape)
print("Size:", array_1d.size)
print("Sum:", array_2d.sum())
数组索引
print("First element:", array_1d[0])
print("Second row, third column:", array_2d[1, 2])
高级功能
NumPy 提供了许多高级功能,如:
- 线性代数运算:解线性方程组、特征值计算等
- 傅里叶变换:用于信号处理
- 随机数生成:用于模拟和实验
更多信息,请参阅 NumPy 官方文档。