numpy.ma.MaskedArray.argmax

方法

ma.MaskedArray.argmax(axis=None, fill_value=None, out=None, *, keepdims=<no value>)

返回沿指定轴的最大值的索引数组。掩码值被视为具有 fill_value 的值。

参数:

axis{None, 整数}

如果为 None，则索引指向扁平化数组，否则沿指定的轴

fill_value标量或 None，可选

用于填充掩码值的数值。如果为 None，则使用 maximum_fill_value(self._data) 的输出。

out{None, 数组}，可选

结果可以放置到的数组。其类型被保留，并且必须具有正确的形状才能容纳输出。

返回值:

index_array{整数数组}

示例

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