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

Question

我有一个numpy的ndarray：

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

无论何时我执行np.argmax(array), 它都不会返回行和列的元组。为什么不会返回？

- Asker123

Answer 1

您没有指定一个轴（axis）。它会返回一个展平数组的索引，如文档所述。

>>> 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)

- wim