将 NumPy 数组转换为 Python 列表结构？

Posted 2023-02-23

【中文标题】将 NumPy 数组转换为 Python 列表结构？

【英文标题】：Converting NumPy array into Python List structure?

【发布时间】：2010-12-30 06:54:24

【问题描述】：

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

【参考方案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 在评论中指出这一点。）

【讨论】：

如果您的列表是 float32 tolist 会将其转换为 floatss - 这可能是不希望的。使用 list(myarray) 并不会因此受到影响 - 为什么我们应该更喜欢 tolist ？

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

我从来没有提到过整数——试试float32_array = np.array([0.51764709], np.float32); print(float32_array.tolist()); print(list(float32_array))

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

沿着这些思路：np.array([[0, 'one'], ['two', 3]]).tolist() -> [['0', 'one'], ['two', '3']]

【参考方案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`

【讨论】：

获取平面列表，还有a.flatten().tolist()（或list(a.flatten()) cf上面的讨论）

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

@deps_stats 这是因为np.array(arr.tolist()) == arr;换句话说，.tolist 是arr 的序列化

【参考方案3】：

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

list(c.flatten())

【参考方案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))

【参考方案5】：

另一种选择

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

也可以。