numpy.concatenate(), 배열 합치기

pythonnumpy

numpy.concatenate() 함수를 이용하여 두 배열을 하나의 배열로 합칠 수 있습니다.

1차원 배열 합치기

numpy.concatenate((A, B))는 배열 A와 B를 하나의 배열로 합칩니다.

import numpy as np

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

arr3 = np.concatenate((arr1, arr2))
print(arr3)

Output:

[1 2 3 4 5 6]

2차원 배열 합치기(행 방향)

numpy.concatenate((A, B), axis=0)은 2차원 배열 A와 B를 행 방향(axis=0)으로 합칩니다.

  • 2차원 배열에서 axis=0은 행 방향 의미

아래 예제의 실행 결과를 보면 두 배열에서 행 단위로 합쳐진 것을 볼 수 있습니다.

import numpy as np

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

arr3 = np.concatenate((arr1, arr2), axis=0)
print(arr3)

Output:

[[1 2]
 [3 4]
 [5 6]]

2차원 배열 합치기(열 방향)

numpy.concatenate((A, B), axis=1)은 2차원 배열 A와 B를 열 방향(axis=1)으로 합칩니다.

  • 2차원 배열에서 axis=1은 열 방향 의미

아래 예제의 실행 결과를 보면 두 배열에서 열 뱡향의 요소들이 하나의 배열로 합쳐진 것을 볼 수 있습니다.

import numpy as np

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

arr3 = np.concatenate((arr1, arr2), axis=1)
print(arr3)

Output:

[[1 2 5]
 [3 4 6]]

다차원 배열 평탄화(1차 배열로 만들기)

numpy.concatenate((A, B), axis=None)은 다차원 배열의 요소들을 모두 합쳐서 1차원 배열로 만드는, 배열 평탄화를 합니다.

axis=None을 입력하지 않으면, 서로 다른 차원을 입력했을 때 예외가 발생할 수 있고 평탄화가 안됩니다.

import numpy as np

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

arr3 = np.concatenate((arr1, arr2), axis=None)
print(arr3)

Output:

[1 2 3 4 5 6]
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha