как применить функцию к массиву numpy
Ответы
Ivan Mamtsev
29 мая 2024
Для применения функции к массиву 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)
Этот код также вернет новый массив, где каждый элемент будет являться квадратом соответствующего элемента из исходного массива.
0
0