numpy reshape 函数的使用
时间: 2023-10-23 19:14:00 浏览: 88
Numpy之reshape()使用详解
numpy中的reshape函数可以将数组转换为不同的形状,其用法如下:
```python
numpy.reshape(a, newshape, order='C')
```
其中,a表示要转换形状的数组,newshape是一个整数或整数元组,表示新的形状,order是可选参数,表示读取元素的顺序。如果不指定order,则默认为C(按行)。
例如,将一个一维数组转换为二维数组:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6])
b = np.reshape(a, (2, 3))
print(b)
```
输出:
```
[[1 2 3]
[4 5 6]]
```
另外,当newshape中有一个轴的值为-1时,reshape函数会自动计算该轴的长度,以满足原始数组和新数组的元素总数相同:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6])
b = np.reshape(a, (2, -1))
print(b)
```
输出:
```
[[1 2 3]
[4 5 6]]
```
阅读全文