[python] NumPy에서 배열의 모양(shape)을 변경하는 방법은 무엇인가요?

NumPy는 파이썬에서 과학적 계산을 위한 핵심 라이브러리이며, 다차원 배열을 다루는데 사용됩니다. 넘파이 배열의 모양(shape)을 변경하는 방법에 대해 알아보겠습니다.

reshape() 함수 사용 reshape() 함수를 사용하여 배열의 모양을 변경할 수 있습니다. reshape() 함수는 배열의 원소 개수를 유지하면서 배열의 차원과 모양을 재구성합니다. 예를 들어, 1차원 배열을 2차원 배열로 변환하거나, 2차원 배열을 3차원 배열로 변환할 수 있습니다.

아래는 NumPy에서 reshape() 함수를 사용하여 배열의 모양을 변경하는 예시 코드입니다.

import numpy as np

# 1차원 배열을 2차원 배열로 변환
arr1d = np.array([1, 2, 3, 4, 5, 6])
arr2d = arr1d.reshape((2, 3))
print(arr2d)
# 출력: [[1 2 3]
#        [4 5 6]]

# 2차원 배열을 3차원 배열로 변환
arr2d = np.array([[1, 2, 3], [4, 5, 6]])
arr3d = arr2d.reshape((2, 3, 1))
print(arr3d)
# 출력: [[[1]
#          [2]
#          [3]]
#
#         [[4]
#          [5]
#          [6]]]

주의사항:

NumPy에서 배열의 모양을 변경하는 방법에 대해 알아보았습니다. reshape() 함수를 사용하여 다양한 모양의 배열을 생성하고 연산에 활용할 수 있습니다.

더 자세한 내용은 NumPy 공식 도큐멘테이션을 참고하세요.