NumPy 배열의 모든 값이 0이 아닌지 확인하는 방법

이 글에서는 NumPy 배열의 모든 값이 0인지 아닌지 확인하는 방법을 알아봅니다. 배열의 모든 요소가 0이 아니라면 결과는 'True'가 되고, 하나라도 0이 포함되어 있다면 'False'가 출력됩니다.

알고리즘을 구현하기 전에 가장 먼저 해야 할 일은 NumPy 라이브러리를 설치하는 것입니다. 명령 프롬프트(터미널)에서 다음 명령어를 실행하면 간단히 설치할 수 있습니다.

pip install numpy

예시

입력:

[1,2,3,4]

출력:

True

입력:

[0,1,2,3]

출력:

False

설명

여기서는 NumPy에서 기본으로 제공하는 내장 함수인 np.all(input_array)를 사용합니다. 이 함수는 배열의 모든 요소를 하나씩 검사하며, 숫자가 0이 아니면 'True'로 평가합니다. 즉, 0이 아닌 모든 요소는 'True'로, 0은 'False'로 처리되며, 배열에 있는 모든 요소가 'True'일 때만 최종적으로 'True'를 반환합니다.

알고리즘

1. 1단계: numpy를 임포트(import)합니다.
2. 2단계: np.array()를 사용해 NumPy 배열을 정의합니다.
3. 3단계: 해당 배열을 np.all() 함수의 매개변수로 전달합니다.
4. 4단계: 결과를 출력하고 종료합니다.

예제 코드

import numpy as np

array1 = np.array([1,2,3,4])
array2 = np.array([0,1,2,3])

print("Array 1: ", array1)
print("Array2: ", array2)
print("\nArray 1 is non-zero: ", np.all(array1))
print("Array 2 is non-zero: ", np.all(array2))

출력

Array 1: [1 2 3 4]
Array2: [0 1 2 3]

Array 1 is non-zero: True
Array 2 is non-zero: False

위 코드에서 첫 번째 배열(array1)은 모든 요소가 0이 아니므로 np.all()이 'True'를 반환하고, 두 번째 배열(array2)에는 0이 포함되어 있으므로 'False'가 반환됩니다. 이처럼 np.all() 함수를 활용하면 복잡한 반복문 없이도 한 줄의 코드로 배열 전체의 조건을 손쉽게 검사할 수 있습니다.