NumPy numpy.ndarray.tolist()的使用
numpy模块提供了一个函数 numpy.ndarray.tolist() ,用于将数组的数据元素转换为列表。此函数将数组作为 a.ndim- 级深度嵌套的Python标量列表返回。
简单来说,该函数将数组元素的副本作为Python列表返回。通过item函数,元素被转换为最接近的兼容的内置Python类型。当’a.ndim’为0时,列表的深度为0,它将是一个简单的Python标量,而不是任何列表。
语法
ndarray.tolist()
参数:
此函数没有参数。
返回值: y: object,或列表的对象,或列表的对象
此函数返回可能是嵌套的数组元素的列表。
注意: 我们可以通过a=np.array(a.tolist())重新创建数组,但是这样做有时会丢失精度。
示例1
如果我们对于一个1D数组使用 a.tolist() ,则与 list(a) 几乎相同,只是 tolist 将numpy标量转换为Python标量。
import numpy as np
a = np.uint32([6, 2])
a
a_list=list(a)
a_list
type(a_list[0])
a_tolist=a.tolist()
a_tolist
type(a_tolist[0])
输出:
array([6, 2], dtype=uint32)
[6, 2]
<type 'numpy.uint32'>
[6L, 2L]
<type 'long'>
在上面的代码中
- 我们使用别名np导入numpy。
- 我们使用np.uint32()函数创建一个数组’a’。
- 我们声明了变量’a_list’,并将 list() 函数的返回值赋给它。
- 我们尝试打印’a’的值,’a_list’的值和a_list的类型。
- 我们声明了变量a_tolist,并将 ndarray.tolist() 的返回值赋给它。
- 最后,我们尝试打印’a_tolist’的类型和值。
在输出中,显示了一个列表和类型,列表的元素是从源数组转换而来的。
示例2
对于一个二维数组,会递归地应用 tolist 。
import numpy as np
a = np.array([[11, 21], [31, 41]])
b=a.tolist()
a
b
输出:
array([[11, 21],
[31, 41]])
[[11, 21], [31, 41]]
在上面的代码中
- 我们使用别名np导入了numpy库。
- 我们使用np.array()函数创建了一个二维数组’a’。
- 我们声明了变量’b’并将其赋值为 a.tolist() 函数的返回值。
- 最后,我们尝试打印值’ b ‘。
在输出中,它显示了一个列表,其中元素是从源数组转换而来的。
示例3
import numpy as np
x = np.array(5)
list(x)
y=x.tolist()
y
输出:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: iteration over a 0-d array
5