NUMPY ПРИМЕНИТЬ ФУНКЦИЮ К МАССИВУ

NumPy – это библиотека Python для научных вычислений, которая включает множество функций для работы с массивами и матрицами. Одной из таких функций является numpy.apply_along_axis, позволяющая применить заданную функцию к массиву по выбранной оси.

Синтаксис функции выглядит следующим образом:

numpy.apply_along_axis(func1d, axis, arr, *args, **kwargs)
func1d - функция для применения к массиву
axis - выбранная ось
arr - массив, к которому нужно применить функцию
*args, **kwargs - дополнительные параметры для функции

Для примера, рассмотрим задачу вычисления среднего арифметического по выбранной оси для трехмерного массива:

import numpy as np
arr = np.random.randint(1, 10, size=(2, 3, 4))
print("Исходный массив:\n", arr)
mean = np.apply_along_axis(np.mean, axis=0, arr=arr)
print("Среднее по оси 0:\n", mean)
mean = np.apply_along_axis(np.mean, axis=1, arr=arr)
print("Среднее по оси 1:\n", mean)
mean = np.apply_along_axis(np.mean, axis=2, arr=arr)
print("Среднее по оси 2:\n", mean)

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

Исходный массив:
[[[1 6 5 4]
[9 4 5 4]
[7 5 9 5]]
[[1 5 9 1]
[1 2 4 2]
[9 7 9 8]]]
Среднее по оси 0:
[[ 1. 5.5 7. 2.5]
[ 5. 3. 4.5 3. ]
[ 8. 6. 9. 6.5]]
Среднее по оси 1:
[[5.66666667 5. 6.33333333 4.33333333]
[3.66666667 4.66666667 7.33333333 3.66666667]]
Среднее по оси 2:
[[4. 5.5 6. 4.5]
[4. 2.25 6. 5.75]]