[파이썬 강의] 69. 모든 값이 1인 배열 원하는 크기로 만들기 (ones)

모든값이 1인 배열을 만들 때는 numpy 모듈의 ones 함수를 사용합니다. 원하는 크기의 배열을 만들 수 있습니다. 

 

1차원 배열 만들기

모든 원소가 1이고, 길이가 10인 1차원 배열을 만들어 봅시다. 

import numpy as np
>>> import numpy as np
>>> np.ones(10)
array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1.])

 

2차원 배열 만들기

모든 값이 1이고, 모양이 3행5열인 2차원 배열을 만들어봅시다. 행과 열의 수를 리스트 형태로 입력합니다. 입력값은 축 개념과 동일합니다. 

>>> np.ones([3,5])
array([[1., 1., 1., 1., 1.],
       [1., 1., 1., 1., 1.],
       [1., 1., 1., 1., 1.]])

3차원 배열 만들기

아래와 같이 입력하면 어떤 배열이 될까요? 

>>> np.ones([3,4,5])

3층이고, 각 층은 4행5열인 배열이 됩니다. 

 

>>> np.ones([3,4,5])
array([[[1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]],

       [[1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]],

       [[1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]]])