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