Функция amax
Функция amax возвращает максимальное значение
элементов массива. Первым параметром функция принимает
массив. Вторым параметром можно указать ось (axis),
вдоль которой производится поиск максимума. Также
функция поддерживает параметр keepdims для сохранения
размерности и параметр initial для задания начального
значения.
Синтаксис
np.amax(a, axis=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>)
Пример
Давайте найдем максимальное значение в одномерном массиве:
import numpy as np
arr = np.array([10, 25, 5, 40, 15])
res = np.amax(arr)
print(res)
Результат выполнения кода:
40
Пример
Давайте найдем максимальное значение в двумерном массиве без указания оси (по всем элементам):
import numpy as np
arr = np.array([[5, 12, 8], [20, 3, 15]])
res = np.amax(arr)
print(res)
Результат выполнения кода:
20
Пример
Найдем максимальные значения вдоль оси 0 (по столбцам):
import numpy as np
arr = np.array([[5, 12, 8], [20, 3, 15]])
res = np.amax(arr, axis=0)
print(res)
Результат выполнения кода:
[20 12 15]
Пример
Найдем максимальные значения вдоль оси 1 (по строкам):
import numpy as np
arr = np.array([[5, 12, 8], [20, 3, 15]])
res = np.amax(arr, axis=1)
print(res)
Результат выполнения кода:
[12 20]
Пример
Используем параметр keepdims для сохранения размерности:
import numpy as np
arr = np.array([[5, 12, 8], [20, 3, 15]])
res = np.amax(arr, axis=0, keepdims=True)
print(res)
print(res.shape)
Результат выполнения кода:
[[20 12 15]]
(1, 3)
Пример
Используем параметр initial для задания начального значения сравнения:
import numpy as np
arr = np.array([10, 20, 30])
res = np.amax(arr, initial=50)
print(res)
Результат выполнения кода:
50
Пример
Используем параметр where для поиска максимума только среди элементов, удовлетворяющих условию:
import numpy as np
arr = np.array([10, 20, 30, 40, 50])
# ищем максимум среди элементов больше 25
mask = arr > 25
res = np.amax(arr, where=mask, initial=0)
print(res)
Результат выполнения кода:
50
