Python –将NumPy数组转换为列表

时间:2020-02-23 14:42:33  来源:igfitidea点击:

我们可以使用numpy ndarray tolist()函数将数组转换为列表。
如果数组是多维的,则返回一个嵌套列表。
对于一维数组,将返回包含数组元素的列表。

要列出的NumPy数组

tolist()函数不接受任何参数。
这是将数组转换为列表表示的一种简单方法。

1.将一维NumPy数组转换为列表

import numpy as np

# 1d array to list
arr = np.array([1, 2, 3])
print(f'NumPy Array:\n{arr}')

list1 = arr.tolist()
print(f'List: {list1}')

输出:

NumPy Array:
[1 2 3]
List: [1, 2, 3]

2.将多维NumPy数组转换为列表

import numpy as np

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

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

list1 = arr.tolist()

print(f'List: {list1}')

输出:

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