행렬(Matrix, Array)에 열(Column) 추가하기

행렬(Matrix, Array)에 열(Column) 추가하기

---

파이썬 버전 : Python 3.6.2

사용 에디터 : PyCharm

  행렬에 열을 추가하는 방법은 정확하게는 추가하고자 하는 열의 개수를 적용한 새로운 행렬을 만든 뒤 데이터를 넣는 방법이라고 설명할 수 있습니다. 이번 글에서는 리스트 형태의 행을 행렬에 추가하는 방법이지만 만약 array의 형태의 행이라면 append 함수를 사용하여 추가해주면 됩니다.

import numpy as np
 
original_data = np.array([[1,2,3],[5,6,7]])
 
new_column = [4,8]
 
new_data = np.zeros((original_data.shape[0],original_data.shape[1]+1))
new_data[:,:-1] = original_data
new_data[:,-1] = new_column

  위의 코드를 실행하면 new_data는 [[1,2,3,4],[5,6,7,8]] 값을 가지고 있는 행렬이 됩니다.