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

简介

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

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