如何在 Python 中将 NumPy 数组转换为列表

NumPy 是一种 Python 的科学计算库，它提供了一种高效的方式来存储和处理大型数组和矩阵。虽然 NumPy 数组在许多情况下用起来比 Python 原生的列表要快，但有时我们需要将 NumPy 数组转换为 Python 列表以便与其他库或框架兼容。

本文将介绍如何在 Python 中将 NumPy 数组转换为列表，并提供一些注意事项。

方法一：使用 tolist()

NumPy 提供了一个函数 tolist()，该函数可以将 NumPy 数组转换为 Python 列表。

示例代码如下：

import numpy as np

arr = np.array([1, 2, 3])
lst = arr.tolist()

print(lst) # Output: [1, 2, 3]

方法二：使用 list()

也可以使用 Python 内置函数 list() 将 NumPy 数组转换为列表。

示例代码如下：

import numpy as np

arr = np.array([1, 2, 3])
lst = list(arr)

print(lst) # Output: [1, 2, 3]

注意事项：

1. NumPy 数组可以是多维数组，如果要将多维数组转换为列表，需要递归地将每个子数组转换为列表。
2. 如果 NumPy 数组中有非基本类型的元素（例如自定义类），则 tolist() 和 list() 方法可能无法正确处理，需要手动转换每个元素。
3. 如果 NumPy 数组是稀疏矩阵，转换为列表可能会丢失原始数据结构的信息。

总结：

转换 NumPy 数组为 Python 列表是一项常见任务。使用 tolist() 和 list() 函数可以轻松地完成此任务，但需要注意上述的注意事项。熟练使用这些函数将有助于您更好地利用 NumPy 库和 Python 语言的强大功能。

火焰兔

海报 链接