[파이썬 numpy] 원하는 크기의 배열 쉽게 만드는 방법

[파이썬 numpy] 

원하는 크기의 배열 쉽게 만드는 방법

배열을 공부할 때 여러 차원의 배열을 정의하며 이래저래 변형해 보고 싶은 경우가 있습니다. 매번 리스트를 이용하여 정의하기가 귀찮은데요. 이럴 때 사용하는 방법입니다. 

arrange 메소드와 reshape 메소드를 함께 사용해주면 됩니다. 몇가지 예시를 통해 알아봅시다. 

2행4열짜리 배열 (shape이 2,4)

>>> A=np.arange(8).reshape(2,4)

>>> A

array([[0, 1, 2, 3],

       [4, 5, 6, 7]])

3행5열에 높이 2짜리 배열 (shape이 2,3,5)

>>> A=np.arange(30).reshape(2,3,5)

>>> A: 

array([[[ 0,  1,  2,  3,  4],

        [ 5,  6,  7,  8,  9],

        [10, 11, 12, 13, 14]],

       [[15, 16, 17, 18, 19],

        [20, 21, 22, 23, 24],

        [25, 26, 27, 28, 29]]])