numpy.ma.MaskedArray.argmax

方法

ma.MaskedArray。argmax ( axis=None , fill_value=None , out=None , * , keepdims=<无 值> )

返回沿给定轴的最大值的索引数组。屏蔽值被视为具有值 fill_value。

参数：

轴{无，整数}

如果为 None，则索引位于展平数组中，否则沿着指定的轴

fill_value标量或 None，可选

用于填充屏蔽值的值。如果没有，则使用 Maximum_fill_value(self._data) 的输出。

out {无，数组}，可选

可以将结果放入其中的数组。它的类型被保留，并且它必须具有正确的形状来保存输出。

返回：

索引数组{整数数组}

例子

>>> a = np.arange(6).reshape(2,3)
>>> a.argmax()
5
>>> a.argmax(0)
array([1, 1, 1])
>>> a.argmax(1)
array([2, 2])