본문 바로가기
Python/NumPy

[초급] NumPy - np.ones(), np.zeros() 사용 방법

by 코린이파이아 2023. 5. 31.

np.ones()

np.ones() 함수에 대해 설명해드릴게요. np.ones() 함수는 주어진 형태(shape)의 배열을 생성하며, 모든 요소를 1로 초기화합니다. 이 함수를 사용하면 원하는 형태의 배열을 쉽게 만들 수 있어요. 예시와 함께 설명해드릴게요.

  1. np.ones() 함수의 입력 형식
    • 입력: np.ones(shape, dtype=None, order='C')
  2. 주요 매개변수 설명
    • shape: 생성할 배열의 형태를 지정합니다. shape은 정수 또는 정수들의 튜플로 입력됩니다.
    • dtype (선택 사항): 배열의 데이터 타입을 지정합니다. 기본값은 None이며, 주어지지 않을 경우 입력된 값에 따라 자동으로 결정됩니다.
    • order (선택 사항): 배열의 메모리 배치 순서를 지정합니다. 기본값은 'C'로, C 스타일의 로우 메이저(row-major) 순서로 배열이 배치됩니다.
  3. np.ones() 함수의 예시
    • 입력: arr = np.ones((2, 3))
      • 출력:
        [[1. 1. 1.] [1. 1. 1.]]
    • 입력: arr = np.ones((3, 2), dtype=int)
      • 출력:
        [[1 1] [1 1] [1 1]]
    • 입력: arr = np.ones((1, 5))
      • 출력:
        [[1. 1. 1. 1. 1.]]
    • 입력: arr = np.ones((2, 2, 2))
      • 출력:
        [[[1. 1.] [1. 1.]] [[1. 1.] [1. 1.]]]

 

 

np.zeros() 

np.zeros() 함수에 대해 설명해드릴게요. np.zeros() 함수는 주어진 형태(shape)의 배열을 생성하며, 모든 요소를 0으로 초기화합니다. 이 함수를 사용하면 원하는 형태의 배열을 손쉽게 만들 수 있어요. 예시와 함께 설명해드릴게요.

  1. np.zeros() 함수의 입력 형식
    • 입력: np.zeros(shape, dtype=None, order='C')
  2. 주요 매개변수 설명
    • shape: 생성할 배열의 형태를 지정합니다. shape은 정수 또는 정수들의 튜플로 입력됩니다.
    • dtype (선택 사항): 배열의 데이터 타입을 지정합니다. 기본값은 None이며, 주어지지 않을 경우 입력된 값에 따라 자동으로 결정됩니다.
    • order (선택 사항): 배열의 메모리 배치 순서를 지정합니다. 기본값은 'C'로, C 스타일의 로우 메이저(row-major) 순서로 배열이 배치됩니다.
  3. np.zeros() 함수의 예시
    • 입력: arr = np.zeros((2, 3))
      • 출력:
        [[0. 0. 0.] [0. 0. 0.]]
    • 입력: arr = np.zeros((3, 2), dtype=int)
      • 출력:
        [[0 0] [0 0] [0 0]]
    • 입력: arr = np.zeros((1, 5))
      • 출력:
        [[0. 0. 0. 0. 0.]]
    • 입력: arr = np.zeros((2, 2, 2))
      • 출력:
        [[[0. 0.] [0. 0.]] [[0. 0.] [0. 0.]]]