将NumPy数组转换为Python列表

Question

如何将一个NumPy数组转换为Python列表？

- Alex Brooks

Answer 1

使用tolist()方法：

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

注意，此操作将从 numpy 类型转换为“最接近兼容的 Python 类型”（列表中）（例如 np.int32 或 np.float32）的值。如果您想保留 numpy 数据类型，可以对数组调用 list() 方法，这样您将得到一个由 numpy 标量 组成的列表。（感谢 Mr_and_Mrs_D 在评论中指出这一点。）

- Peter Hansen

---

如果你的列表是float32类型，tolist方法将把它转换成floats类型——这可能不是我们想要的。而使用list(myarray)则没有这个问题——为什么我们应该更喜欢使用tolist呢？ - Mr_and_Mrs_D

@Mr_and_Mrs_D，您是在建议尽管数组中有浮点数，但您希望得到的结果列表是整数？那就是完全不同的问题了...当然，您必须指定尽管是float32值，它们都将是整数。无论如何，如果您遇到这种情况，我想执行myarray.astype(np.int32).tolist()将是一个选项，并且明确说明您要完成的任务。(最后，当我尝试时，list(array_of_type_float32)在这里并不会给出整数...所以我不知道您在问什么。) - Peter Hansen

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

好的，所以其中一个把数据类型转换为[float]，而另一个仍使用numpy标量float32类型，即[np.float32]。知道了。很好，但我猜它是否可取取决于每个具体的情况。值得一提的是，我怀疑通常当有人（像OP一样）要求将数据转换为列表时，他们隐含地指的是普通Python数据类型的列表，即在本例中为浮点数或整数的列表，而不是numpy标量类型的列表。感谢您指出这一点：我已编辑答案并包含了这方面的说明。 - Peter Hansen

沿着这条线：np.array([[0, 'one'], ['two', 3]]).tolist() -> [['0', 'one'], ['two', '3']] - keithpjolley

@Mr_and_Mrs_D 对于上述问题的回答是，当我们想要将数据序列化为json或yaml文件时，应该使用 tolist。 - off99555

Answer 2

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

list(c.flatten())

- Harshal SG

Answer 3

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

- CodingMatters

---

要获得一个平坦的列表，还可以使用 a.flatten().tolist()（或者 list(a.flatten())，参见上面的讨论）。 - ClementWalter

@ClementWalter 谢谢！那个评论比答案本身更有价值。你知道为什么.tolist()会创建一个嵌套列表吗？为什么我们要经过这么多步骤才能得到一个简单的列表？这背后的原理是什么？ - deps_stats

@deps_stats 这是因为 np.array(arr.tolist()) == arr；换句话说，.tolist 是 arr 的一种序列化形式。 - ClementWalter

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

- Shivid

Answer 5

另一个选择

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]

同样适用。

- rahul-ahuja

Answer 6

最简单的将数组转换为列表的方法是使用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)

- Charchit Shukla