python把ndarray转为列表

介绍

把numpy.ndarray类型转为列表类型。

使用numpy.ndarray.flatten()方法

import numpy as np

# 创建一个n维数组（例如，3维数组）
n_dim_array = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

# 使用flatten方法将其转换为一维数组
flat_array = n_dim_array.flatten()

flat_list = flat_array.tolist()

print(flat_list)  # 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

使用列表推导式（对于非numpy数组）

# 创建一个嵌套的列表（模拟n维数组）  
nested_list = [[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]  
  
# 使用列表推导式将其转换为一维列表  
flat_list = [item for sublist in nested_list for subsublist in sublist for item in subsublist]  
  
print(flat_list)  # 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]