numpy.ma.masked_array.argmax#

方法

ma.masked_array.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])