NumPy 是 Python 的一个强大的科学计算库,它允许你创建各种类型的数组。以下是一些常见的 NumPy 数组生成方法:
- 使用 numpy.array() 函数:这是最常用的 NumPy 数组生成方法。它接受一个列表或其他序列作为参数,并将其转换为 NumPy 数组。
例如:
import numpy as np
arr = np.array([1, 2, 3])
print(arr)
输出结果:
[1, 2, 3]
- 使用 numpy.zeros() 函数:该函数创建一个全零的数组。
例如:
arr = np.zeros((3, 3))
print(arr)
输出结果:
[[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.]]
- 使用 numpy.ones() 函数:该函数创建一个全一的数组。
例如:
arr = np.ones((3, 3))
print(arr)
输出结果:
[[1., 1., 1.],
[1., 1., 1.],
[1., 1., 1.]]
- 使用 numpy.linspace() 函数:该函数创建一个均匀分布的数组。
例如:
arr = np.linspace(0, 10, 5)
print(arr)
输出结果:
[0. 2.5 5. 7.5 10. ]
- 使用 numpy.arange() 函数:该函数创建一个等差数列。
例如:
arr = np.arange(5)
print(arr)
输出结果:
[0, 1, 2, 3, 4]
- 使用 numpy.eye() 函数:该函数创建一个对角线上为 1 ,其余为 0 的二维数组。
例如:
arr = np.eye(3)
print(arr)
输出结果:
[[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]]
以上只是 NumPy 数组生成方法的一部分,还有更多的方法,请参考 NumPy 官方文档进行学习。