본문 바로가기
Python/numpy

파이썬 numpy.apply_along_axis 함수 활용하기

by PySun 2024. 8. 31.
반응형

Numpy apply_along_axis 함수 소개:

Numpy의 `apply_along_axis` 함수는 특정 함수를 배열의 특정 축을 따라 적용하여 결과를 반환하는 데 사용됩니다. 이 함수를 사용하면 다차원 배열의 각 행 또는 열에 대해 사용자가 지정한 함수를 적용할 수 있습니다.

기본 사용법:

`apply_along_axis` 함수를 사용하여 배열의 특정 축을 따라 함수를 적용하는 방법은 다음과 같습니다.

import numpy as np

# 적용할 함수 정의
def my_func(x):
    return np.sum(x)  # 예시로 각 행 또는 열의 합을 반환하는 함수

# 다차원 배열 생성
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# apply_along_axis 함수를 사용하여 함수 적용
result = np.apply_along_axis(my_func, axis=0, arr=arr)

print("함수 적용 결과:", result)

위의 예시 코드에서는 `apply_along_axis` 함수를 사용하여 각 열의 합을 계산합니다. 적용할 함수를 정의하고, 이를 `apply_along_axis` 함수에 전달하여 배열의 각 열에 함수를 적용한 결과를 얻습니다.

결과:

`apply_along_axis` 함수를 사용하면 다차원 배열의 특정 축을 따라 함수를 적용하여 유연하게 처리할 수 있습니다. 위의 예시 코드를 참고하여 `apply_along_axis` 함수를 활용해보세요.

반응형