[파이썬][numpy] numpy `swapaxes` 함수

Numpy는 파이썬에서 수치 연산을 위한 강력한 라이브러리로, 다차원 배열을 처리하는 데 매우 유용합니다. 이 중에서 swapaxes 함수는 배열의 축을 교환하여 데이터를 재구성하는 데 사용됩니다. 이번 블로그 포스트에서는 swapaxes 함수의 사용법과 예제를 살펴보겠습니다.

swapaxes 함수 개요

swapaxes 함수는 다차원 배열의 축을 교환합니다. 이를 통해 배열의 차원을 변경하거나 데이터의 형태를 바꿀 수 있습니다. swapaxes 함수는 다음과 같은 형태로 사용됩니다:

numpy.swapaxes(arr, axis1, axis2)

예제: 배열의 축 교환하기

다음은 swapaxes 함수의 간단한 예제 코드입니다. 이 예제에서는 3차원 배열의 축을 교환하여 데이터를 재구조화합니다.

import numpy as np

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

# 축 교환하기: axis 0 (깊이)와 axis 2 (열)을 교환
swapped_arr = np.swapaxes(arr, 0, 2)

# 결과 출력
print("원래 배열:\n", arr)
print("\n축 교환 후 배열:\n", swapped_arr)

실행 결과:

원래 배열:
 [[[ 1  2  3]
  [ 4  5  6]]

 [[ 7  8  9]
  [10 11 12]]]

축 교환 후 배열:
 [[[ 1  7]
  [ 4 10]]

 [[ 2  8]
  [ 5 11]]

 [[ 3  9]
  [ 6 12]]]

요약

Numpy의 swapaxes 함수는 다차원 배열에서 축을 교환하여 데이터를 재구조화하는 데 사용됩니다. 이를 통해 배열의 차원을 변경하거나 데이터 형태를 바꿀 수 있습니다. 이번 포스트에서는 swapaxes 함수의 개요와 사용법을 설명하였으며, 예제 코드를 통해 실제 활용 방법을 알아보았습니다.

더 자세한 내용 및 다른 Numpy 함수에 대해서는 Numpy 공식 문서를 참고하시기 바랍니다.