如何将NumPy数组转换为Python列表(例如[[1,2,3],[4,5,6]]
),并且速度相当快?
发布于 2009-12-27 23:31:19
使用tolist()
import numpy as np
>>> np.array([[1,2,3],[4,5,6]]).tolist()
[[1, 2, 3], [4, 5, 6]]
请注意,这会将值从它们可能具有的任何numpy类型(例如np.int32或np.float32)转换为“最近兼容的Python类型”(在列表中)。如果想保留numpy scalars数据类型,可以在数组上调用list(),最终得到一个numpy列表。(感谢Mr_and_Mrs_D在评论中指出了这一点。)
发布于 2016-09-25 09:49:57
如果numpy数组形状是2D,则numpy .tolist方法会生成嵌套列表。
如果需要平面列表,可以使用下面的方法。
import numpy as np
from itertools import chain
a = [1,2,3,4,5,6,7,8,9]
print type(a), len(a), a
npa = np.asarray(a)
print type(npa), npa.shape, "\n", npa
npa = npa.reshape((3, 3))
print type(npa), npa.shape, "\n", npa
a = list(chain.from_iterable(npa))
print type(a), len(a), a`
发布于 2020-06-05 12:29:56
C= np.array([1,2,3,4,5,6])
list(c.flatten())
https://stackoverflow.com/questions/1966207
复制相似问题