[Python/파이썬] numpy.hstack() 및 numpy.vstack()로 배열 연결하기

numpy.hstack() 및 numpy.vstack() 함수는 배열을 수평으로 또는 수직으로 연결하는 데 사용됩니다. 이 두 함수를 사용하여 배열을 연결하는 방법을 살펴보겠습니다.

numpy.hstack() 함수 예제

numpy.hstack() 함수는 주어진 배열들을 수평으로 연결하여 하나의 배열로 만듭니다. 아래는 이 함수의 사용 예제입니다.

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = np.hstack((arr1, arr2))
print(result)

위의 코드는 두 개의 1차원 배열을 수평으로 연결하여 하나의 배열로 만듭니다. numpy.hstack() 함수에 연결할 배열들을 튜플 형태로 전달하여 사용합니다.

numpy.vstack() 함수 예제

numpy.vstack() 함수는 주어진 배열들을 수직으로 연결하여 하나의 배열로 만듭니다. 아래는 이 함수의 사용 예제입니다.

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6]])
result = np.vstack((arr1, arr2))
print(result)

위의 코드는 두 개의 2차원 배열을 수직으로 연결하여 하나의 배열로 만듭니다. numpy.vstack() 함수를 사용하여 두 배열을 수직으로 연결합니다.

추가 예제

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6]])
arr3 = np.array([[7, 8], [9, 10]])
result = np.hstack((arr1, arr2, arr3))
print(result)

위의 코드는 세 개의 2차원 배열을 수평으로 연결하여 하나의 배열로 만듭니다. numpy.hstack() 함수를 사용하여 세 배열을 수평으로 연결합니다.