如何从python中的4d矩阵切片？

Question

让我们说我有一个numpy形状为ParseRequestURI的4维矩阵A。我想玩第二个维度，即4。我有一个二维数组[2,4,5,5]。

我希望输出是这样的：

idx=[[1,2],[0,3]]，即B= A[:,idx,:,:]的形状必须为B。

关于如何有效实现这一目标的任何建议？