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

作者 Pankaj

导言

在 NumPy 中，[np.array] 对象可以通过 tolist() 函数转换为一个列表。tolist()函数不接受任何参数。如果数组是一维的，将返回一个包含数组元素的列表。对于多维数组，则返回一个嵌套的列表。

将一维 NumPy 数组转换为列表

让我们构造一个一维数组 [1, 2, 3]：

import numpy as np

# 一维数组转换为列表
arr_1 = np.array([1, 2, 3])

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

以上代码会输出：

NumPy Array:
[1 2 3]

现在，我们来使用 tolist()：

import numpy as np

# 一维数组转换为列表
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

# 二维数组转换为列表
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

# 二维数组转换为列表

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 对象转换为列表。tolist() 适用于一维和多维数组。