NumPy apply_along_axis()

1 개요

NumPy apply_along_axis()

numpy.apply_along_axis()

import numpy as np

def my_func(a):
    """1차원 배열에서 첫번째와 마지막 원소의 평균"""
    return (a[0] + a[-1]) * 0.5

b = np.array([[1,2,3], [4,5,6], [7,8,9]])

np.apply_along_axis(my_func, 0, b)

np.apply_along_axis(my_func, 1, b)

b = np.array([[8,1,7], [4,3,9], [5,2,6]])
np.apply_along_axis(sorted, 1, b)

b = np.array([[1,2,3], [4,5,6], [7,8,9]])
np.apply_along_axis(np.diag, -1, b)

2 같이 보기

• NumPy foreach row
• NumPy foreach column

3 참고

• https://numpy.org/doc/stable/reference/generated/numpy.apply_along_axis.html