将一维数组np.arange(10)转换为2行的二维数组
时间: 2024-03-05 20:47:52 浏览: 117
Python创建二维数组实例(关于list的一个小坑)
可以使用 numpy 的 `reshape` 函数将一维数组转换为指定形状的二维数组,具体代码如下:
```python
import numpy as np
arr = np.arange(10)
new_arr = arr.reshape(2, -1) # -1表示自动计算另一个维度的大小
print(new_arr)
```
输出结果为:
```
[[0 1 2 3 4]
[5 6 7 8 9]]
```
其中,`reshape(2, -1)` 表示将数组转换为 2 行的二维数组,而 `-1` 表示自动计算另一个维度的大小,这里计算得到的大小为 5,使得数组的总大小不变。
阅读全文