如何在numpy中找到二维数组的argmax？

Question

我有一个numpy ndarray：

array = np.array([[1,2,3],[4,5,6],[7,8,9]])

每当我做np.argmax(array)时，它都不会返回行和列的元组。  为什么不呢？

Answer 1

您没有指定axis。它将一个索引返回到展平数组as documented。

>>> L = [[1,2,3],[4,5,6],[7,8,9]]
>>> np.argmax(L)
8
>>> np.array(L).ravel()[np.argmax(L)]
9

如果要从该索引获取行和列，请将其解析：

>>> np.unravel_index(8, np.array(L).shape)
(2, 2)