python 将两个一维数组变为二维数组
时间: 2023-12-29 18:04:27 浏览: 288
python将三维数组展开成二维数组的实现
可以使用numpy库中的reshape函数将两个一维数组变为二维数组。假设两个一维数组分别为a和b,要将其组合成一个2行n列的二维数组,可以使用如下代码:
```python
import numpy as np
a = np.array([1, 2, 3, 4])
b = np.array([5, 6, 7, 8])
c = np.array([a, b])
d = c.reshape(2, -1)
print(d)
```
输出结果为:
```
[[1 2 3 4]
[5 6 7 8]]
```
其中reshape函数的第一个参数为行数,第二个参数为列数。为了方便,可以将列数设置为-1,表示自动计算列数。
阅读全文