将NumPy数组转换为列表可以使用tolist()函数,该函数是NumPy数组对象的一个方法。
示例代码1:
import numpy as np
# 创建一个二维NumPy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 转换为列表
list_arr = arr.tolist()
print("NumPy数组:\n", arr)
print("转换为列表:\n", list_arr)
输出结果:
NumPy数组:
[[1 2 3]
[4 5 6]]
转换为列表:
[[1, 2, 3], [4, 5, 6]]
示例代码2:
import numpy as np
# 创建一个一维NumPy数组
arr = np.array([1, 2, 3, 4, 5])
# 转换为列表
list_arr = arr.tolist()
print("NumPy数组:\n", arr)
print("转换为列表:\n", list_arr)
输出结果:
NumPy数组:
[1 2 3 4 5]
转换为列表:
[1, 2, 3, 4, 5]
以上示例代码中,我们首先导入NumPy模块并创建一个NumPy数组对象arr,然后使用arr.tolist()方法将对象转换为列表,最后打印出转换前后的数组和列表。注意到tolist()方法可以用于所有NumPy数组类型,包括多维数组和一维数组。