NumPy array()用法详解

在 NumPy 中，可以使用 array() 来创建一个数组。array() 是 NumPy 中最基础也是最常用的一个方法。

array() 函数的语法格式如下：

np.array(列表或元组)

array() 的参数可以是一个列表，也可以是一个元组。

array()其实就是将一个列表转换为一个数组，或者将一个元组转换为一个数组。

【实例】将列表转换为数组

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(arr)
print(type(arr))
print(arr.dtype)

输出结果如下：

[1 2 3 4 5]
<class 'numpy.ndarray'>
int32

从输出结果可以看出，数组元素之间是用空格隔开的，而不是用逗号隔开的，这一点和列表不一样。dtype 属性用于获取数组元素的类型。

array() 会自动根据“列表元素的类型”来确定“数组元素的数据类型”。np.array() 返回的是一个 ndarray 对象。

可能一些读者会问：为什么要将列表转换成数组？原因很简单，对于列表，只能使用列表的方法来对其中的元素进行操作，但是将列表转换成数组之后，就可以使用数组提供的丰富方法来对其中的元素进行操作了。

【实例】将元组转换为数组

import numpy as np
arr = np.array((1, 2, 3, 4, 5))
print(arr)
print(type(arr))
print(arr.dtype)

输出结果如下：

[1 2 3 4 5]
<class "numpy.ndarray">
int32

np.array() 除了可以将列表转换成数组，还可以将元组转换成数组。不过在实际开发中，我们更多是将一个列表转换成数组。