numpy.ma.argmax#

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

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

参数:

axis{None, integer}: 如果为 None，则索引是到展平的数组，否则是沿指定的轴。
fill_valuescalar or None, optional: 用于填充掩码值的数值。如果为 None，则改用 maximum_fill_value(self._data) 的输出。
out{None, array}, optional: 结果可以放置的数组。其类型将被保留，并且它必须具有正确的形状来容纳输出。

返回:

示例

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