numpy.ma.MaskedArray.argmax#

方法

ma.MaskedArray.argmax(axis=None, fill_value=None, out=None, *, keepdims=<no value>)[源代码]#

返回沿给定轴的最大值的索引数组.被掩码的值被视为具有填充值.

参数:
axis{None, 整数}

如果为 None,索引是针对展平的数组,否则沿指定的轴进行.

fill_value标量或无,可选

用于填充掩码值的值.如果为 None,则使用 maximum_fill_value(self._data) 的输出代替.

out{None, array}, 可选

可以放置结果的数组.它的类型被保留,并且它必须具有正确的形状以容纳输出.

返回:
index_array{integer_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])