Python numpy MaskedArray.argmax用法及代码示例

本文简要介绍 python 语言中 numpy.ma.MaskedArray.argmax 的用法。

用法:

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

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

参数：

axis： {无，整数}

如果为 None，则索引在展平数组中，否则沿指定轴

fill_value： 标量或无，可选

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

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