博客

将NumPy数组转换为列表是Python数据分析中非常常见的操作之一。下面是将NumPy数组转换为列表的完整攻略：

1. 使用tolist()函数

将NumPy数组转换为列表的最简单方法是使用tolist()函数。tolist()函数是NumPy中的一个函数，用于将数组转换为列表。

示例代码：

import numpy as np

# 构造一个NumPy数组
arr = np.array([1, 2, 3, 4, 5])

# 使用tolist()函数将NumPy数组转换为列表
lst = arr.tolist()

print(lst)  # 输出 [1, 2, 3, 4, 5]

1. 使用tolist()函数和ndarray.flatten()函数

如果NumPy数组是多维数组，可以使用tolist()函数和ndarray.flatten()函数结合起来进行转换。这种方法可以将多维数组转换为一维列表。

示例代码：

import numpy as np

# 构造一个NumPy多维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用tolist()函数将NumPy数组转换为嵌套列表
lst = arr.tolist()

# 使用ndarray.flatten()函数将嵌套列表转换为一维列表
lst = arr.flatten().tolist()

print(lst)  # 输出 [1, 2, 3, 4, 5, 6, 7, 8, 9]

以上就是将NumPy数组转换为列表的完整攻略，希望能帮助你解决相关问题。