4d numpy数组的多维索引

Question

我有两个3d体积堆叠在尺寸为2x5x5x5的4d阵列中。

如果要获取单个3d体积的某些/所有体素的值，则可以使用三个向量并获得以下值：

x_new_mesh, y_new_mesh, z_new_mesh = np.meshgrid(np.arange(m),
                                                 np.arange(n),
                                                 np.arange(p))

x_new = x_new_mesh.flatten()
y_new = y_new_mesh.flatten()
z_new = z_new_mesh.flatten()
values = vol[x_new, y_new, z_new]

m，n，p = 5且x，y，z为1x125向量。我得到了一个1x125的阵列，没有任何问题。

但是，我没有设法对堆积的卷做类似的事情。如何使用两组索引（每个卷具有不同的索引）来获得具有所需值的2x125数组？

Answer 1

假设vol是您的2x5x5x5数组：

values = vol[:, x_new, y_new, z_new]