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

297

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

python numpy

425

用途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标量列表。（感谢Mr_and_Mrs_D在评论中指出这一点。）

— 彼得·汉森
source

6

如果您的列表为float32，tolist则会将其转换为floatss-可能不希望如此。使用list(myarray)并不会因此而受苦-我们为什么要喜欢tolist呢？

— Mr_and_Mrs_D

1

@Mr_and_Mrs_D您是否建议，尽管数组中有浮点数，但您希望结果列表为整数吗？那是一个完全不同的问题……当然，您必须指定尽管是float32值，但它们都将是不可或缺的。无论如何，如果您遇到这种情况，我想这样做myarray.astype(np.int32).tolist()是一种选择，并且明确说明您要完成的工作。（最后，list(array_of_type_float32)当我尝试时不给整数...所以我不知道你在问什么。）

— Peter Hansen

1

我从未提到整数-试试float32_array = np.array([0.51764709], np.float32); print(float32_array.tolist()); print(list(float32_array))

— Mr_and_Mrs_D

2

好的，所以一个转换为[float]，另一个仍然使用numpy标量float32类型，例如[np.float32]。精细。很高兴知道，但是我想这是否可取取决于每个具体案例。对于它的价值，我怀疑通常是当某人（例如OP）要求转换为列表时，它们隐含表示常规Python数据类型的列表（在这种情况下为浮点数或整数），而不是numpy标量类型的列表。感谢您指出这一点：我已经编辑了答案，其中包括关于此的注释。

— 彼得·汉森

np.array([[0, 'one'], ['two', 3]]).tolist() -> [['0', 'one'], ['two', '3']]

— 遵循

11

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

— 有抱负的上师
source

5

tolist()熊猫说，即使遇到嵌套数组，也可以正常工作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))

— 希维德
source

0

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

— 加内什·贾达夫（Ganesh Jadhav）
source

0

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

list(c.flatten())

— harshal SG
source