python二维数组怎么转化为一维数组

1242
2023/10/12 3:54:26
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

要将一个二维数组转化为一维数组,可以使用列表推导式(list comprehension)或者使用numpy库中的flatten()函数。

方法1: 使用列表推导式

# 示例二维数组
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# 使用列表推导式将二维数组转化为一维数组
new_arr = [element for sublist in arr for element in sublist]
print(new_arr)  # 输出: [1, 2, 3, 4, 5, 6, 7, 8, 9]

方法2: 使用numpy库的flatten()函数

import numpy as np
# 示例二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 使用flatten()函数将二维数组转化为一维数组
new_arr = arr.flatten()
print(new_arr)  # 输出: [1 2 3 4 5 6 7 8 9]

注意:如果使用numpy库的flatten()函数,需要先安装numpy库(可以通过pip install numpy命令进行安装)。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: python中main函数的用法是什么