首先,可以用 np.array 从 Python 列表创建数组:
# 整型数组:
np.array([1, 4, 2, 5, 3])
得到如下结果:
array([1, 4, 2, 5, 3])
请记住,不同于 Python 列表,NumPy 要求数组必须包含同一类型的数据。如果类型不匹配,NumPy 将会向上转换(如果可行)。这里整型被转换为浮点型:
np.array([3.14, 4, 2, 3])
得到如下结果:
array([ 3.14, 4. , 2. , 3. ])
如果希望明确设置数组的数据类型,可以用 dtype 关键字:
np.array([1, 2, 3, 4], dtype='float32')
得到如下结果:
array([ 1., 2., 3., 4.], dtype=float32)
最后,不同于 Python 列表,NumPy 数组可以被指定为多维的。以下是用列表的列表初始化多维数组的一种方法:
# 嵌套列表构成的多维数组
np.array([range(i, i + 3) for i in [2, 4, 6]])
得到如下结果:
array([[2, 3, 4],
[4, 5, 6],
[6, 7, 8]])
内层的列表被当作二维数组的行。