python把ndarray转为列表

  • 介绍
  • 使用numpy.ndarray.flatten()方法
  • 使用列表推导式(对于非numpy数组)

介绍

把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]
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐