使用沿给定维度的(n-1)维数组访问n维数组的最优雅方法是什么,如虚拟示例中所示
a = np.random.random_sample((3,4,4))
b = np.random.random_sample((3,4,4))
idx = np.argmax(a, axis=0)
我现在如何访问idx a以获得最大值,a就像我使用过一样a.max(axis=0)？或者如何检索idxin中指定的值b？
我想过使用,np.meshgrid但我认为这是一种矫枉过正.请注意,尺寸axis可以是任何有用的轴(0,1,2),并且事先不知道.有一种优雅的方式来做到这一点？