numpy.matrix.argmax #

方法

矩阵。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]])