博客

将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数组类型，包括多维数组和一维数组。