python将一个三维方阵添加0变成一个四维方阵
时间: 2024-03-01 09:53:34 浏览: 15
可以使用numpy库中的pad函数对三维方阵进行0填充,然后使用reshape函数改变方阵的维度。
示例代码:
```python
import numpy as np
# 原始三维方阵
matrix = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
# 在三维方阵周围添加一圈0,变为 4x4x3 的方阵
padded_matrix = np.pad(matrix, ((1,1),(1,1),(1,1)), mode='constant', constant_values=0)
# 改变方阵维度为 2x2x4x4
reshaped_matrix = padded_matrix.reshape(2, 2, 4, 4)
print(reshaped_matrix)
```
输出结果:
```
[[[[0 0 0 0]
[0 1 2 0]
[0 3 4 0]
[0 0 0 0]]
[[0 0 0 0]
[0 5 6 0]
[0 7 8 0]
[0 0 0 0]]]
[[[0 0 0 0]
[0 0 0 0]
[0 0 0 0]
[0 0 0 0]]
[[0 0 0 0]
[0 0 0 0]
[0 0 0 0]
[0 0 0 0]]]]
```