python二维数组转一维
时间: 2023-07-11 17:59:16 浏览: 90
将二维数组转为一维数组的2种方法
5星 · 资源好评率100%
可以使用numpy库中的`flatten()`函数将二维数组转换为一维数组,示例代码如下:
```python
import numpy as np
# 二维数组
arr_2d = np.array([[1, 2], [3, 4]])
# 转换为一维数组
arr_1d = arr_2d.flatten()
print(arr_1d) # 输出 [1 2 3 4]
```
除此之外,也可以使用列表推导式或者for循环实现二维数组转一维数组的操作,示例代码如下:
```python
# 使用列表推导式
arr_1d = [num for row in arr_2d for num in row]
# 使用for循环
arr_1d = []
for row in arr_2d:
for num in row:
arr_1d.append(num)
print(arr_1d) # 输出 [1, 2, 3, 4]
```
阅读全文