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

介绍

With NumPy, np.array对象可以转换为列表tolist()功能。这tolist()函数不接受任何参数。如果数组是一维的，则返回包含数组元素的列表。对于多维数组，返回嵌套列表。

先决条件

为了完成本教程，您将需要：

• 熟悉 Python 3 的安装和使用pip安装软件包。并熟悉 Python 编码。如何使用 Python 3 进行编码系列或使用适用于 Python 的 VS 代码.

本教程已使用 Python 3.9.6 和 NumPy 1.23.3 进行测试。

将一维 NumPy 数组转换为列表

我们来构造一个一维数组[1, 2, 3]:

import numpy as np

# 1d array to list
arr_1 = np.array([1, 2, 3])

print(f'NumPy Array:\n{arr_1}')

该代码将输出：

NumPy Array:
[1 2 3]

现在，让我们使用tolist():

import numpy as np

# 1d array to list
arr_1 = np.array([1, 2, 3])

print(f'NumPy Array:\n{arr_1}')

list_1 = arr_1.tolist()

print(f'List: {list_1}')

这个新代码将输出：

List: [1, 2, 3]

该数组已转换为numpy标量到 Python 标量。

将多维 NumPy 数组转换为列表

我们来构建一个多维数组[ [1, 2, 3], [4, 5, 6] ]:

import numpy as np

# 2d array to list
arr_2 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'NumPy Array:\n{arr_2}')

该代码将输出：

NumPy Array:
[[1 2 3]
 [4 5 6]]

现在，让我们使用tolist():

import numpy as np

# 2d array to list
arr_2 = np.array([[1, 2, 3], [4, 5, 6]])

print(f'NumPy Array:\n{arr_2}')

list_2 = arr_2.tolist()

print(f'List: {list_2}')

这个新代码将输出：

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

该数组已转换为numpy标量到 Python 标量。

结论

在本文中，您学习了如何使用tolist()转换np.array对象列表。它适用于一维和多维数组。

参考

• API文档