对于三维数组,生成一个新数组与之维度相同,但长度为2倍,python代码
时间: 2024-05-15 13:14:34 浏览: 97
python将三维数组展开成二维数组的实现
5星 · 资源好评率100%
可以使用numpy库来生成一个新的数组,其代码如下:
``` python
import numpy as np
# 生成一个三维数组
a = np.zeros((2, 2, 2))
# 生成一个新数组与之维度相同,但长度为2倍
b = np.zeros((2, 2, 4))
# 循环遍历三维数组,将值赋给新数组
for i in range(2):
for j in range(2):
for k in range(2):
b[i][j][2*k] = a[i][j][k]
b[i][j][2*k+1] = a[i][j][k]
print(b)
```
以上代码会先生成一个三维数组a,然后根据a的维度生成一个新的三维数组b,但长度为原来的2倍。最后使用循环将a的值赋给b,因为每个元素的长度变为2倍,所以需要在新数组中重复存储。
阅读全文