파이썬, Numpy 배열 합치기

파이썬, Numpy 배열 합치기

 

글. 수알치 오상문

 

다음 예제는 리스트를 넘파이 배열로 만들고 합치는 예제입니다.

 

import numpy as np  # 넘파이 모듈 가져오기

array1 = np.array([1, 2, 3])   # 리스트를 넘파이 1차원 배열로 저장
array2 = np.array([4, 5, 6])   # 리스트를 넘파일 1차원 배열로 저장

array3 = np.concatenate([array1, array2])   # 1차원 배열 합치기
print(array3.shape)   # 배열 구조 출력
print(array3)             # 배열 출력

 

print('---------------------------')

넘파이 1차원 배열을 2차원 배열로 변경하여 저장

array4 = array1.reshape(1, 3)  # [[1,2,3]]
array5 = array2.reshape(1, 3)  # [[4,5,6]]

# 2차원 배열을 위아래로 합치기

array6 = np.concatenate([array4, array5], axis=0)
print(array6.shape)
print(array6)

 

print('---------------------------')

array7 = np.concatenate([array4, array5], axis=1)
print(array7.shape)
print(array7)

 

[실행 결과]

(6,)
[1 2 3 4 5 6]
---------------------------
(2, 3)
[[1 2 3]
[4 5 6]]
---------------------------
(1, 6)
[[1 2 3 4 5 6]]

 

<이상>