numpy 행렬 나누기

import numpy as np

matrix = np.array([[ 0, 1, 2, 3],
                   [ 4, 5, 6, 7],
                   [ 8, 9,10,11], 
                   [12,13,14,15]])

numpy에서 행렬을 쪼깰 때는 split을 사용한다. 

예제 matrix를 행 중심으로 나눠, 

a 변수에는 1행 ~ 3행의 행렬이, b 변수에는 4행의 행렬이 들어가도록 하려면, 행 지정과 axis=0을 이용한다.

 

c 변수에는 1열 ~ 3열, d 변수에는 4열의 행렬이 들어가도록 하려면, 열 지정과 axis=1을 이용한다.