展平二维数组 Python

在Python中，展平二维数组是一个非常常见的操作，它可以将一个二维数组转换为一维数组，方便后续的处理。本文将介绍Python中如何展平二维数组。

方法一：使用列表推导式

arr = [[1, 2], [3, 4], [5, 6]]
flat_arr = [item for sublist in arr for item in sublist]
print(flat_arr)

输出结果：

[1, 2, 3, 4, 5, 6]

列表推导式是Python中常用的语法糖，可以非常简洁地进行列表转换操作。上面的代码中，我们使用了两个for循环来遍历二维数组中的每一个元素，将其添加到一维数组中。

方法二：使用numpy库

import numpy as np
arr = [[1, 2], [3, 4], [5, 6]]
flat_arr = np.array(arr).flatten()
print(flat_arr)

输出结果：

[1 2 3 4 5 6]

numpy是Python中非常流行的数学运算库，它提供了很多方便的函数来操作数组。其中，flatten函数可以将多维数组转换为一维数组。上面的代码中，我们将二维数组转换为了numpy中的矩阵，然后使用flatten函数将其展平。

方法三：使用itertools库

import itertools
arr = [[1, 2], [3, 4], [5, 6]]
flat_arr = list(itertools.chain.from_iterable(arr))
print(flat_arr)

输出结果：

[1, 2, 3, 4, 5, 6]

itertools是Python中处理迭代器的标准库，其中的chain函数可以将多个迭代器串联起来。from_iterable函数则可以从一个可迭代对象中获取元素，这里我们使用from_iterable函数将二维数组转换为一维数组。