[Python] np.ones np.zeros - 1 혹은 0으로 이뤄진 배열 생성

본 포스팅은 numpy를 활용하여 1 혹은 0으로 이뤄진 배열을 구성하는 법에 대해 포스팅하겠다.

 

np.ones()

특정 사이즈를 기준으로 one으로 이뤄진 배열을 만드는 함수이다. 

 

입력 변수로는 생성하고자 하는 크기를 입력하면 되며, 1차원 배열을 구성하고자 할 때 그 길이에 해당하는 숫자 하나만 작성해 줘도 되며, 만일 2차원 이상의 배열을 생성하고 싶다면 크기를 tuple이나 list 형식으로 지정해주면 된다.

 

import numpy as np

print(np.ones(3))
# array([1., 1., 1.])


print(np.ones([2,3]), np.ones((2,3)))
"""
(array([[1., 1., 1.],
        [1., 1., 1.]]),
 array([[1., 1., 1.],
        [1., 1., 1.]]))
"""

print(np.ones([2,3,2]))
"""
[[[1. 1.]
  [1. 1.]
  [1. 1.]]

 [[1. 1.]
  [1. 1.]
  [1. 1.]]]
"""

 

np.zeros()

np.zeros()는 np.ones()와 다르게 특정 사이즈를 기준으로 zero로 이뤄진 배열로 만들어준다.

 

사용방법은 np.ones()과 동일하다.

 

print(np.zeros(3))
# [0. 0. 0.]


print(np.zeros([2,3]), np.zeros((2,3)))
"""
[[0. 0. 0.]
 [0. 0. 0.]] [[0. 0. 0.]
 [0. 0. 0.]]
"""


print(np.zeros([2,3,2]))
"""
[[[0. 0.]
  [0. 0.]
  [0. 0.]]

 [[0. 0.]
  [0. 0.]
  [0. 0.]]]
"""

 

 

 

Ref : https://numpy.org/doc/stable/reference/generated/numpy.ones.html, https://numpy.org/doc/stable/reference/generated/numpy.zeros.html