如何将一个一维和二维的ndarray转换为列表

Ling_Ze

于 2024-03-05 09:51:32 发布

阅读量322

点赞数 3

CC 4.0 BY-SA版权

分类专栏： python数据处理文章标签： python

本文链接：https://blog.youkuaiyun.com/Ling_Ze/article/details/136470233

python数据处理专栏收录该内容

40 篇文章

订阅专栏

本文详细介绍了如何使用NumPy库中的tolist(),ravel(),和flatten()方法将一维和二维ndarray转换为Python列表，以及如何处理多维数组的扁平化过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 如何将一个一维和二维的ndarray转换为列表
import numpy as np

# 创建一个一维的ndarray
arr_1d = np.array([1, 2, 3, 4, 5])

# 使用tolist()将一维ndarray转换为列表
list_1d = arr_1d.tolist()
print(list_1d)  # 输出: [1, 2, 3, 4, 5]

# 创建一个二维的ndarray
arr_2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用tolist()将二维ndarray转换为列表
list_2d = arr_2d.tolist()
print(list_2d)  # 输出: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 如果你的ndarray是多维的，并且你希望得到一个完全扁平化的列表（即，所有维度都被展开成一个一维列表），
# 你可以使用ravel()或flatten()方法，然后再调用tolist()
flattened_list = arr_2d.flatten().tolist()
print(flattened_list)  # 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9]

flattened_list = arr_2d.ravel().tolist()
print(flattened_list)  # 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9]```