如何将numpy数组转化为list|极客教程

如何将numpy数组转化为list

在Python中，NumPy是一个非常流行的数值计算库，可以执行各种数学运算和操作数组。然而，有时候我们需要将NumPy数组转化为Python原生的列表（list），以便进行一些特定的操作或者与其他库进行交互。本文将详细介绍如何将NumPy数组转化为列表，并提供一些示例代码和运行结果。

方法一：使用tolist()方法

NumPy数组对象有一个内置的tolist()方法，可以直接将数组转化为列表。这种方法非常简单且有效。

import numpy as np

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

# 将NumPy数组转化为列表
arr_list = arr.tolist()

print(arr_list)

运行结果如下：

[[1, 2, 3], [4, 5, 6]]

如上所示，使用tolist()方法可以直接将NumPy数组转换为列表，每一个子数组对应列表中的一个元素，并且保持原有的结构。

方法二：使用tolist()函数

另一种常见的方法是使用tolist()函数，它可以将任何可迭代对象转化为列表。这种方法也可以用于将NumPy数组转化为列表。

import numpy as np

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

# 使用tolist()函数将NumPy数组转化为列表
arr_list = np.tolist(arr)

print(arr_list)

运行结果如下：

[1, 2, 3, 4, 5]

可以看到，使用tolist()函数同样可以将NumPy数组转化为列表，但是需要注意的是这种方法只能用于一维数组。

方法三：使用tolist()函数和tolist()方法结合

如果我们想要将多维数组转化为列表，可以结合使用tolist()函数和tolist()方法。首先使用tolist()函数将NumPy数组转化为Python原生的嵌套列表，然后再使用tolist()方法将每个子数组转化为列表。

import numpy as np

# 创建一个二维NumPy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

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

# 使用tolist()方法将嵌套列表转化为列表
arr_list = [subarr.tolist() for subarr in nested_list]

print(arr_list)

运行结果如下：