php中文网

如何高效读取NumPy数组中的数据?

php中文网

如何读取 numpy.ndarray 中的数据

想要读取 numpy.ndarray 中的数据,可以使用 tolist() 方法。

tolist() 方法可以将高维度的 ndarray 递归地转换为列表嵌套结构。对于一维的 ndarray,tolist() 与 list(a) 的效果相同。

例如,对于二维数组 a:

>>> import numpy as np
>>> a = np.array([[1, 2], [3, 4]])
>>> a.tolist()
[[1, 2], [3, 4]]

tolist() 方法可以递归地将数组中的每个元素转换为列表。

如果数组是零维的,使用 list(b) 会引发错误,但 b.tolist() 不会出现问题。

>>> b = np.array(1)
>>> b.tolist()
1
>>> list(b)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: iteration over a 0-d array
>>>

以上就是如何高效读取NumPy数组中的数据?的详细内容,更多请关注php中文网其它相关文章!