python - 了解 argmax

Question

假设我有矩阵

import numpy as np    
A = np.matrix([[1,2,3,33],[4,5,6,66],[7,8,9,99]])

我试图理解函数 argmax，据我所知它返回最大值

如果我在 Python 上尝试:

np.argmax(A[1:,2])

我应该获取第二行中的最大元素直到行尾(即第三行)和第三列吗？所以它应该是数组 [6 9]，arg max 应该返回 9？但为什么当我在 Python 上运行它时，它返回值 1？

如果我想返回从第 2 行开始到第 3 列的最大元素(即 9)，我应该如何修改代码？

我已经查看了 Python 文档，但还是有点不清楚。感谢您的帮助和解释。

Answer 1

没有 argmax 返回最大值的位置。 max 返回最大值。

import numpy as np    
A = np.matrix([[1,2,3,33],[4,5,6,66],[7,8,9,99]])

np.argmax(A)  # 11, which is the position of 99

np.argmax(A[:,:])  # 11, which is the position of 99

np.argmax(A[:1])  # 3, which is the position of 33

np.argmax(A[:,2])  # 2, which is the position of 9

np.argmax(A[1:,2])  # 1, which is the position of 9