假设一个数组维度为[10, 4, 1024],想要依次取出第二维度的4个数组,取出的数组形状为[10, 1, 1024]代码该怎么写
时间: 2024-04-12 10:29:45 浏览: 31
二维数组实例代码
你可以使用切片操作来实现这个功能。根据你的需求,可以使用以下代码:
```python
import numpy as np
# 假设数组名为arr,维度为[10, 4, 1024]
arr = np.random.random((10, 4, 1024))
# 使用切片操作取出第二维度的4个数组,并改变形状为[10, 1, 1024]
result = arr[:, 1:2, :]
print(result.shape) # 输出结果为(10, 1, 1024)
```
在上述代码中,`arr[:, 1:2, :]`表示取出数组arr的所有第一维度的元素(即所有的10个数组),第二维度取索引为1的元素(即第二个数组),并保持第三维度不变。最终得到的结果形状为[10, 1, 1024]。
阅读全文