在 Python 中将 NumPy 数组转换为列表

列表和数组是 Python 中两个最基本且最常用的集合对象。

它们都是可变的，用于以通用名称存储元素的集合，并且每个元素都有可用于访问它的特定位置。

但是，有一些显着差异。列表已经在 Python 中内置，而对于数组，我们需要导入 arrays 或 NumPy 模块，并在使用它们之前声明数组。数组还可以更有效地将数据存储在内存中，并被广泛用于数学运算。

在本教程中，我们将把一个 numpy 数组转换为一个列表。

使用 tolist() 方法将一个 NumPy 数组转换为列表

NumPy 数组的 tolist() 方法可以将 NumPy 数组转换为列表。

例如，

import numpy as np

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

print(oned.tolist())
print(twod.tolist())

输出：

[[1, 2, 3]]
[[1, 2, 3], [4, 5, 6]]

请注意，此方法将整个数组视为一个元素。这就是为什么当我们将其与二维数组一起使用时，它会返回列表列表。

为了避免这种情况，我们可以将 tolist() 与 flatten() 或 ravel() 方法一起使用，它们可以将 N-D 数组转换为线性一维数组。这两种方法执行相同的功能。不同之处在于 ravel() 方法返回数组的参考视图并影响原始数组，而 flatten() 方法对数组的副本起作用。因此，ravel() 函数被认为速度更快，占用的内存更少。

以下代码显示了这些函数的用法。

import numpy as np

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

print(oned.flatten().tolist())
print(twod.flatten().tolist())


print(oned.ravel().tolist())
print(twod.ravel().tolist())

输出：

[1, 2, 3]
[1, 2, 3, 4, 5, 6]
[1, 2, 3]
[1, 2, 3, 4, 5, 6]

使用 for 循环将 Python 中的 NumPy 数组转换为列表

对于那些刚接触编程或想要定制最终列表的人来说，这只是一种基本方法。我们遍历一个数组，并将每个元素分别附加到一个空列表中。以下代码实现了这一点。

import numpy as np

arr = np.array([1, 2, 3])
lst = []

for x in arr:
    lst.append(x)
    
print(lst)

输出：

[1, 2, 3]

火焰兔

海报 链接