Question

如何将NumPy数组转换为Python列表（例如[[1,2,3],[4,5,6]]），并且速度相当快？

Answer 1

使用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数据类型，你可以在你的数组上调用list（），最后你会得到numpy scalars的列表。（感谢Mr_and_Mrs_D在评论中指出这一点。）

Answer 2

如果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`

Answer 3

c = np.array（[[1,2,3]，[4,5,6]]）

list(c.flatten())

Answer 4

tolist()即使遇到嵌套数组也能正常工作，例如pandas DataFrame;

my_list = [0,1,2,3,4,5,4,3,2,1,0]
my_dt = pd.DataFrame(my_list)
new_list = [i[0] for i in my_dt.values.tolist()]

print(type(my_list),type(my_dt),type(new_list))

Answer 5

另一种选择

c.ravel()
# or
c.ravel().tolist()

也有效。

Answer 6

someList = [list(map(int, input().split())) for i in range(N)]

将NumPy数组转换为Python List结构？

6 个答案: