Python numpy matrix.argmax用法及代碼示例

本文簡要介紹 python 語言中 numpy.matrix.argmax 的用法。

用法:

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

沿軸的最大值的索引。

返回沿指定軸第一次出現最大值的索引。如果軸為無，則索引用於展平矩陣。

參數：

See `numpy.argmax` for complete descriptions：

注意：

這與 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]])