Numpy数组转换为列表

Numpy数组转换为列表

在Python中，我们可以使用Numpy库来处理数组，我们需要将Numpy数组转换为Python列表，以下是一些方法来实现这个功能。

方法1：使用tolist()方法

Numpy数组对象提供了一个tolist()方法，可以将数组转换为列表。

示例代码：

import numpy as np
创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
使用tolist()方法将numpy数组转换为列表
list_arr = arr.tolist()
print(list_arr)

输出结果：

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

方法2：使用列表推导式

我们可以使用列表推导式将Numpy数组的每个元素转换为列表。

示例代码：

import numpy as np
创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
使用列表推导式将numpy数组转换为列表
list_arr = [list(row) for row in arr]
print(list_arr)

输出结果：

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

方法3：使用numpy的ravel()方法

Numpy数组有一个ravel()方法，可以将多维数组展平为一维数组，我们可以使用tolist()方法将其转换为列表。

示例代码：

import numpy as np
创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
使用ravel()方法将numpy数组展平为一维数组
flat_arr = arr.ravel()
使用tolist()方法将一维数组转换为列表
list_arr = flat_arr.tolist()
print(list_arr)

输出结果：

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