如何将NumPy数组转换为Python列表?


当前回答

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))

其他回答

将数组转换为列表最简单的方法是使用numpy包:

import numpy as np
#2d array to list
2d_array = np.array([[1,2,3],[8,9,10]])
2d_list = 2d_array.tolist()

要检查数据类型,可以使用以下方法:

type(object)

另一个选择

c = np.array([[1,2,3],[4,5,6]])

c.ravel()
#>> array([1, 2, 3, 4, 5, 6])

# or
c.ravel().tolist()
#>> [1, 2, 3, 4, 5, 6]

同样适用。

c = np.array([[1,2,3],[4,5,6]])

list(c.flatten())

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))

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