nump 求二维数组最大值索引

a = np.random.randn(3, 3)
print(a)
print(np.unravel_index(np.argmax(a), a.shape))

# output
# [[ 0.70479033 -1.43807352  2.36603598]
# [ 1.44277125 -0.73066114  0.9084934 ]
# [ 1.25677773  1.43417866 -0.52670885]]
# (0, 2)

np.argmax默认会将数组a展平，这样的得到的索引为一维的位置，所以利用unravel_index函数将其恢复到数组a对应的位置，因此传入a.shape。