如何获得最后 2 个轴的 argmax？ [复制]

【问题标题】：How to get the argmax of last 2 axes? [duplicate]如何获得最后 2 个轴的 argmax？ [复制]
【发布时间】：2020-05-29 15:30:42
【问题描述】：

假设我有一个数组，它的维度是10 X 20 X 12 X 12 X 2 X 2。如果我想获得最后两个轴的最大值索引该怎么办？意思是我想要一个函数，它将输出（如果所描述的数组是输入）一个形状数组：10 X 20 X 12 X 12。所以（数组是被描述的数组）：

array[indices] == np.amax(array, (-2,-1))

【问题讨论】：

【解决方案1】：

一个简单的解决方案是调用该函数两次，每个轴一次：

A = np.arange(1000).reshape((10,10,10))
np.amax(np.amax(A, axis=-1), axis=-1)

对于np.argmax，您基本上可以这样做。但当然，指数正在发生变化。您需要对此进行更正。

【讨论】：