python将二维数组变为一维数组
时间: 2023-08-09 10:04:47 浏览: 89
详解Python二维数组与三维数组切片的方法
5星 · 资源好评率100%
你可以使用numpy库中的flatten方法将二维数组变为一维数组。例如:
```python
import numpy as np
# 创建一个二维数组
arr_2d = np.array([[1, 2], [3, 4]])
# 使用flatten方法将二维数组变为一维数组
arr_1d = arr_2d.flatten()
print(arr_1d) # 输出 [1 2 3 4]
```
你也可以使用reshape方法将二维数组变为一维数组。例如:
```python
import numpy as np
# 创建一个二维数组
arr_2d = np.array([[1, 2], [3, 4]])
# 使用reshape方法将二维数组变为一维数组
arr_1d = arr_2d.reshape(-1)
print(arr_1d) # 输出 [1 2 3 4]
```
这里的`-1`表示自动计算数组的长度,即根据原始数组的维度来自动计算新的数组长度。
阅读全文