Numpy array 를 shape로 차원확인 하기

연산을 수행하다보면, 동적으로 할당된 array 에 대해서 몇행, 몇열 행렬로 구성되었는지 알아야 할 경우가 있다. 

이 경우 array의 shape 를 이용하여 간단하게 알 수 있다.  


a = np.array( [ [1,2,3],[4,5,6],[7,8,9],[10,11,12] ])
# a 는 3 x 3 행렬로 구성됨.

print(a.shape)

print 결과는 (4,3 ) 임.

추가로 shape[0], shape[1]를 이용하여 전체 행의 갯수와 열의 갯수를 반환받을 수 있다.

즉 a.shape[0] 결과는 4임(shape 튜플의 첫번째 요소는 4) , a.shape[1]의 결과는 3임(shape 튜플의 두번째 요소는 3).