[파이썬][Numpy] 배열 형태(shape) 변경하기

24 Aug 2023

numpy

Numpy 배열의 형태(shape)를 변경하는 방법을 설명하는 예제를 제공하겠습니다. 배열의 형태를 변경하면 데이터의 구조를 조정할 수 있으며, 이는 데이터 전처리나 다양한 연산에 유용합니다.

import numpy as np

# 배열 생성 예제
arr = np.array([1, 2, 3, 4, 5, 6])

# 배열의 형태 변경
reshaped_arr = arr.reshape(2, 3)  # 2행 3열의 2차원 배열로 형태 변경
print("배열 형태 변경:")
print(reshaped_arr)

# 다차원 배열을 1차원 배열로 평탄화 (flatten)
flattened_arr = reshaped_arr.flatten()
print("\n1차원 배열로 평탄화:")
print(flattened_arr)

# 배열 전치 (Transpose)
transposed_arr = reshaped_arr.T  # 행과 열을 바꾸어 전치
print("\n배열 전치:")
print(transposed_arr)

위의 예제에서는 다음과 같은 형태 변경 작업을 수행합니다:

주어진 배열을 생성합니다.
reshape 함수를 사용하여 배열의 형태를 변경하여 2행 3열의 2차원 배열로 만듭니다.
flatten 함수를 사용하여 다차원 배열을 1차원 배열로 평탄화시킵니다.
T 속성을 사용하여 배열을 전치시킵니다.

실행 결과는 다음과 같을 것입니다:

배열 형태 변경:
[[1 2 3]
 [4 5 6]]

1차원 배열로 평탄화:
[1 2 3 4 5 6]

배열 전치:
[[1 4]
 [2 5]
 [3 6]]

이렇게 Numpy를 사용하면 배열의 형태를 유연하게 조절할 수 있으며, 데이터 분석 및 처리 작업에서 필요한 형태로 배열을 가공할 수 있습니다.