как применить функцию к массиву numpy

Для применения функции к массиву NumPy можно воспользоваться функцией np.vectorize. Например, предположим, у нас есть массив NumPy:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])

И нам нужно применить функцию np.square к каждому элементу этого массива. Мы можем сделать это следующим образом:

result = np.vectorize(np.square)(arr)

Результатом будет новый массив, где каждый элемент будет являться квадратом соответствующего элемента из исходного массива.

Также можно использовать лямбда функции и функцию np.apply_along_axis для применения функции к массиву. Например:

result = np.apply_along_axis(lambda x: x ** 2, axis=0, arr=arr)

Этот код также вернет новый массив, где каждый элемент будет являться квадратом соответствующего элемента из исходного массива.