python二维数组转一维
时间: 2023-07-06 14:39:52 浏览: 83
二维数组转一维数组
在Python中,可以使用numpy库来进行二维数组和一维数组之间的转换。具体做法如下:
假设现在有一个2×3的二维数组arr:
```
arr = [[1, 2, 3],
[4, 5, 6]]
```
要将它转换为一维数组,可以使用numpy的`reshape()`函数,将arr变成一个1×6的一维数组:
```
import numpy as np
arr = np.array(arr) # 先将列表转换为numpy数组
arr_1d = arr.reshape(1, -1) # 将arr转换为1x6的一维数组
arr_1d = arr_1d.flatten() # 将1x6的数组展平为一维数组
print(arr_1d) # 输出[1 2 3 4 5 6]
```
其中,`reshape()`函数的参数中,第一个参数表示转换后数组的行数,第二个参数表示列数。如果传入-1,则表示该维度的大小由函数自动计算得出。
最后,使用`flatten()`函数将多维数组展平为一维数组。
阅读全文