要将一个二维数组转化为一维数组,可以使用列表推导式(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函数的用法是什么