求一个数组中某个值（最大值）索引的方法

两种方法：

#method1
np.argwhere(a == np.max(a))

#method2
np.unravel_index(a.argmax(), a.shape)

此方法不是很好，如果有多个最大值，只能返回第一个
展示：


code:

import numpy as np
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9],[3, 6, 8, 7]])
a
print(np.argmax(a))
print(np.argmax(a, axis=0))  #竖着比较，返回行号
print(np.argmax(a, axis=1))  #横着比较，返回列号

np.unravel_index(a.argmax(), a.shape)

np.argwhere(a == np.max(a))