python 三维形状为[1,4,4]的数组 改为形状为[4,4]的数组 代码
时间: 2023-11-22 18:54:40 浏览: 46
python将三维数组展开成二维数组的实现
5星 · 资源好评率100%
可以使用numpy库的reshape函数来改变数组的形状。代码如下:
```python
import numpy as np
# 创建一个三维形状为[1,4,4]的数组
arr_3d = np.array([[[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16]]])
# reshape成[4,4]的数组
arr_2d = arr_3d.reshape(4, 4)
print(arr_2d)
```
运行结果:
```
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]
[13 14 15 16]]
```
可以看到,通过reshape函数将三维数组转换为二维数组。
阅读全文