获取多维NumPy数组中最大值的位置

Question

如何获得多维NumPy数组中最大值的位置（索引）

Answer 1

您只需编写一个函数（仅适用于2d）：

def argmax_2d(matrix):
    maxN = np.argmax(matrix)
    (xD,yD) = matrix.shape
    if maxN >= xD:
        x = maxN//xD
        y = maxN % xD
    else:
        y = maxN
        x = 0
    return (x,y)