numpy 的reshape
时间: 2023-10-10 09:14:38 浏览: 52
Numpy之reshape()使用详解
Numpy的reshape函数可以用于将数组转换为特定的行数和列数。具体来说,reshape函数接受两个参数:数组和目标的行数和列数。通过指定这些参数,我们可以重新排列数组的元素,以满足新的形状要求。例如,通过调用reshape函数,我们可以将一个一维数组转换为一个二维数组,也可以将一个二维数组转换为一个三维数组。
举个例子,假设我们有一个一维数组 a = [1, 2, 3, 4, 5, 6],我们可以使用reshape函数将其转换为一个二维数组,如下所示:
```python
import numpy as np
a = [1, 2, 3, 4, 5, 6]
b = np.reshape(a, (2, 3))
print(b)
```
输出结果为:
```
[[1 2 3]
[4 5 6]]
```
在这个例子中,reshape函数将一维数组 a 重新排列为一个2行3列的二维数组。
除了可以将数组转换为二维数组,reshape函数还可以用于将数组转换为更高维度的矩阵。例如,我们可以使用reshape函数将多个二维数组堆叠在一起,从而创建一个三维数组。
```python
import numpy as np
a = [1, 2, 3]
b = [4, 5, 6]
c = [7, 8, 9]
w = np.array([[a, b, c], [a, b, c]])
print(w)
```
输出结果为:
```
[[
阅读全文