[Python/파이썬] Numpy flatten() 함수, 1차원 배열로 변환하기(평탄화)

Numpy의 `flatten()` 함수는 다차원 배열을 1차원 배열로 평탄화하는 데 사용됩니다. 이 함수를 사용하여 다차원 배열을 1차원 배열로 변환하는 방법에 대해 알아보겠습니다.

예제 1: 2차원 배열 평탄화

가장 간단한 경우는 2차원 배열을 1차원 배열로 변환하는 것입니다. 아래 예제에서는 이를 확인할 수 있습니다.

import numpy as np

# 2차원 Numpy 배열 생성
my_array = np.array([[1, 2, 3], [4, 5, 6]])

# 1차원 배열로 변환
flattened_array = my_array.flatten()

print(flattened_array)

위의 코드에서는 `flatten()` 함수를 사용하여 2차원 Numpy 배열인 `my_array`를 1차원 배열인 `flattened_array`로 변환합니다.

예제 2: 다차원 배열 평탄화

다차원 배열도 마찬가지로 `flatten()` 함수를 사용하여 평탄화할 수 있습니다. 아래 예제에서는 다차원 배열을 1차원 배열로 변환하는 방법을 살펴봅니다.

import numpy as np

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

# 1차원 배열로 변환
flattened_array = my_array.flatten()

print(flattened_array)

위의 코드에서는 다차원 Numpy 배열인 `my_array`를 `flatten()` 함수를 사용하여 1차원 배열로 변환합니다.