reshape函数将三维数组改为二维
时间: 2023-10-23 20:23:11 浏览: 92
python将三维数组展开成二维数组的实现
reshape函数是用来改变数组的形状的。如果你想将一个三维数组改为二维数组,可以使用reshape函数来实现。具体操作如下:
```python
import numpy as np
# 创建一个三维数组
arr = np.array([[[1, 2, 3], [4, 5, 6]],
[[7, 8, 9], [10, 11, 12]]])
# 使用reshape函数将三维数组改为二维数组
reshaped_arr = arr.reshape((arr.shape * arr.shape), arr.shape)
print("原始数组:\n", arr)
print("改变形状后的数组:\n", reshaped_arr)
```
在上面的例子中,我们首先创建了一个三维数组`arr`。然后使用reshape函数将`arr`转换成了一个二维数组`reshaped_arr`。通过指定目标形状`(arr.shape * arr.shape), arr.shape`,我们将原始数组的前两个维度合并为新数组的第一个维度,而保持原来的第三个维度不变。
希望这个例子可以帮助到你!如果你还有其他问题,请随时提问。
阅读全文