numpy.ma.MaskedArray.argmax

方法

ma.MaskedArray.argmax(axis=None, fill_value=None, out=None)[源代码]

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

参数
axis{NO,整数}

如果没有,则索引将进入扁平数组,否则沿指定轴

fill_value标量或无,可选

用于填充屏蔽值的值。如果没有,则使用最大填充值(self.u数据)的输出。

out无,数组,可选

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

返回
index_array{integer_array}

实例

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