NumPy apply_along_axis()

1 개요

NumPy apply_along_axis()

numpy.apply_along_axis()

Python

Reload

Copy

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)

Loading

Copy

np.apply_along_axis(my_func, 1, b)

Loading

Copy

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

Loading

Copy

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

Loading

2 같이 보기

• NumPy foreach row
• NumPy foreach column

3 참고

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