본문 바로가기
Python/numpy

(numpy) 차원 관련 함수 (np.shape, np.size, np.reshape)

by re-hwi 2022. 10. 14.

np.shape

: 매개변수로 받은 array가 몇 행 몇 열인지 튜플 형식으로 반환한다. 

 

예시) 

import numpy as np

arr = np.array([[1,2,3],[4,5,6]])
print(arr.shape)

>>> (2, 3) 

2행 3열 반환

arr 의 형태는 [[1,2,3],   

                       [4,5,6]] 의 형태로도 쓸 수 있으며 shape는 열과 행의 개수를 반환하는 함수이지 인덱스를 반환하지 않는다

np.size

: 매개변수 array의 원소의 개수를 반환한다. 

 

예시)

import numpy as np

arr = [[1,2,3],[4,5,6]]
print(np.size(arr))

>>> 6 

 

np.reshape (x,y)

: array의 차원과 모양을 매개변수로 받은 x행 y열로 바꿔준다. 이 때 x,y의 개수가 원소의 개수보다 많거나 적다면 오류가 발생하니 주의

 

예시)

import numpy as np

array = np.array([1,2,3,4,5,6])
print(array.reshape(3,2))

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

※ 이 때 reshape(3,1)이나 (3,3)으로 입력한다면 원소의 개수와 맞지 않아 오류가 발생

 

 

반응형

댓글