介绍
使用 NumPy, np.array 对象可以转换为具有 tolist() 函数的列表. tolist() 函数不接受任何参数. 如果数组是单维的,则返回包含数组元素的列表. 对于多维数组,返回嵌入列表。
前提条件
为了完成本教程,您将需要:
- 熟悉安装 Python 3 并使用
pip来安装软件包,并熟悉在 Python 中编码(How to Code in Python 3 series)或使用 VS Code for Python。
本教程是用 Python 3.9.6 和 NumPy 1.23.3 测试的。
将单维NumPy Array转换为列表
让我们构建一个单维的数组 [1, 2, 3]:
1import numpy as np
2
3# 1d array to list
4arr_1 = np.array([1, 2, 3])
5
6print(f'NumPy Array:\n{arr_1}')
此代码将输出:
1NumPy Array:
2[1 2 3]
现在,让我们使用‘tolist()’:
1import numpy as np
2
3# 1d array to list
4arr_1 = np.array([1, 2, 3])
5
6print(f'NumPy Array:\n{arr_1}')
7
8list_1 = arr_1.tolist()
9
10print(f'List: {list_1}')
这个新代码将输出:
1List: [1, 2, 3]
该数组已被转换为numpy尺度为Python尺度。
将多维 NumPy Array 转换为列表
让我们构建一个多维的数组 [ [1, 2, 3], [4, 5, 6]:
1import numpy as np
2
3# 2d array to list
4arr_2 = np.array([[1, 2, 3], [4, 5, 6]])
5
6print(f'NumPy Array:\n{arr_2}')
此代码将输出:
1NumPy Array:
2[[1 2 3]
3 [4 5 6]]
现在,让我们使用‘tolist()’:
1import numpy as np
2
3# 2d array to list
4arr_2 = np.array([[1, 2, 3], [4, 5, 6]])
5
6print(f'NumPy Array:\n{arr_2}')
7
8list_2 = arr_2.tolist()
9
10print(f'List: {list_2}')
这个新代码将输出:
1List: [[1, 2, 3], [4, 5, 6]]
该数组已被转换为numpy尺度为Python尺度。
结论
在本文中,您了解如何使用 tolist() 将 np.array 对象转换为列表. 它适用于单维和多维数组。
参考 *