[python] NumPy 배열의 형태 변경하기

14 Dec 2023

NumPy는 파이썬의 과학 및 수학 연산을 위한 핵심 라이브러리로, 다차원 배열을 다루는데 유용합니다. 이번 포스트에서는 NumPy 배열의 형태를 변경하는 방법에 대해 알아보겠습니다.

1. `reshape` 함수

NumPy의 reshape 함수는 배열의 모양을 변경할 수 있습니다. 이 함수는 새로운 모양의 배열을 반환하며, 원래의 배열을 수정하지 않습니다.

예를 들어, 1차원 배열을 2차원 배열로 변경하려면 다음과 같이 reshape 함수를 사용할 수 있습니다:

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
reshaped_arr = arr.reshape(2, 3)
print(reshaped_arr)

위 코드는 [1, 2, 3, 4, 5, 6]의 1차원 배열을 2행 3열의 2차원 배열로 변경합니다.

flatten 함수는 다차원 배열을 1차원 배열로 펼치는 역할을 합니다.

예를 들어, 2차원 배열을 1차원 배열로 변경하려면 다음과 같이 flatten 함수를 사용할 수 있습니다:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
flattened_arr = arr.flatten()
print(flattened_arr)

위 코드는 [[1, 2, 3], [4, 5, 6]]의 2차원 배열을 [1, 2, 3, 4, 5, 6]의 1차원 배열로 변경합니다.

NumPy 배열의 형태를 쉽게 변경할 수 있는 reshape 함수와 flatten 함수를 잘 사용하면 다양한 배열 연산을 보다 유연하게 수행할 수 있습니다.