温馨提示×

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

小亿
120
2023-09-27 03:54:26
栏目: 编程语言

要将一个二维数组转化为一维数组,可以使用列表推导式(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命令进行安装)。

0