numpy.ma.MaskedArray.argmax

MaskedArray.argmax(axis=None, fill_value=None, out=None) [source]

Возвращает массив индексов максимальных значений по заданной оси. Маскированные значения обрабатываются так, как если бы они имели значение fill_value.

Параметры:

axis : {None, целое число} Если None, индекс относится к уплощенному массиву, в противном случае — по указанной оси fill_value : {переменная}, необязательно Значение, используемое для заполнения маскированных значений. Если None, используется результат maximum_fill_value(self._data). out : {None, массив}, необязательно Массив, в который можно поместить результат. Его тип сохраняется, и он должен быть правильной формы, чтобы содержать выходные данные.

Возвращает:

index_array : {массив целых чисел}

Примеры

>>> a = np.arange(6).reshape(2,3)
>>> a.argmax()
5
>>> a.argmax(0)
array([1, 1, 1])
>>> a.argmax(1)
array([2, 2])