Функция fmax

Функция fmax сравнивает два массива поэлементно и возвращает новый массив, содержащий максимальные значения из каждой пары элементов. Если один из сравниваемых элементов является NaN, функция возвращает значение другого элемента, не являющегося NaN. Если оба элемента равны NaN, результатом будет NaN. Первым параметром передается первый массив, вторым параметром - второй массив. Третьим параметром можно указать массив, в который будет сохранен результат.

Синтаксис

np.fmax(x1, x2, out=None)

Пример

Сравним два одномерных массива чисел и найдем максимальные значения поэлементно:

import numpy as np

arr1 = np.array([1, 5, 3, 8])
arr2 = np.array([4, 2, 6, 3])
res = np.fmax(arr1, arr2)
print(res)

Результат выполнения кода:

[4 5 6 8]

Пример

Рассмотрим работу функции с числами NaN. Функция fmax игнорирует NaN и возвращает числовое значение из пары, если оно присутствует:

import numpy as np

arr1 = np.array([1, np.nan, 3, np.nan])
arr2 = np.array([4, 2, np.nan, 6])
res = np.fmax(arr1, arr2)
print(res)

Результат выполнения кода:

[ 4. 2. 3. 6.]

Пример

Сравним два двумерных массива одинаковой формы:

import numpy as np

arr1 = np.array([[1, 2, 3], [4, 5, 6]])
arr2 = np.array([[6, 5, 4], [3, 2, 1]])
res = np.fmax(arr1, arr2)
print(res)

Результат выполнения кода:

[[6 5 4]
 [4 5 6]]

Пример

Используем параметр out для сохранения результата в заранее подготовленный массив:

import numpy as np

arr1 = np.array([10, 20, 30, 40])
arr2 = np.array([40, 30, 20, 10])
out_arr = np.empty(4, dtype=int)
np.fmax(arr1, arr2, out=out_arr)
print(out_arr)

Результат выполнения кода:

[40 30 30 40]

Пример

Функция поддерживает вещание (broadcasting). Сравним массив с числом:

import numpy as np

arr1 = np.array([1, 3, 2, 5, 4])
res = np.fmax(arr1, 3)
print(res)

Результат выполнения кода:

[3 3 3 5 4]

Смотрите также

функцию maximum,
которая также находит поэлементный максимум, но возвращает NaN, если он присутствует в любом из элементов
функцию fmin,
которая находит поэлементный минимум, игнорируя NaN
функцию minimum,
которая находит поэлементный минимум, но чувствительна к NaN
функцию absolute,
которая вычисляет абсолютные значения элементов массива