numpy.matrix.argmax

方法

matrix.argmax(axis=None, out=None)

沿轴的最大值的索引。

返回沿指定轴上的最大值的第一次出现位置的索引。如果 axis 为 None，则索引是针对扁平化矩阵的。

参数：

参见 `numpy.argmax` 获取完整描述

另请参见

numpy.argmax

备注

这与 ndarray.argmax 相同，但返回一个 matrix 对象，而 ndarray.argmax 将返回一个 ndarray。

示例

>>> x = np.matrix(np.arange(12).reshape((3,4))); x
matrix([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
>>> x.argmax()
11
>>> x.argmax(0)
matrix([[2, 2, 2, 2]])
>>> x.argmax(1)
matrix([[3],
        [3],
        [3]])