불리언 배열이 numpy.sum 함수에서 어떻게 더해지나?

불리언 배열은 참(True) 또는 거짓(False) 값을 가지는 1차원 배열입니다. 그리고 numpy.sum은 배열의 원소들을 더해주는 함수인데, 불리언 배열을 numpy.sum 함수에 인자로 전달하면 어떻게 될까요? 이번 글에서는 예제를 통해서 불리언의 수학적 연산에 대해 알아보겠습니다.

 

불리언에 대한 기본적인 내용은 아래 링크를 참조하세요.

Python 기초: 불리언(Boolean)과 비교 연산자

Python 기초: 불리언(Boolean)과 비교 연산자

 

불리언의 True와 False를 숫자로 하면?

불리언 배열을 numpy 함수들에 인자로 전달하면 True는 1로, False는 0으로 간주됩니다.

 

1) numpy.sum 함수에서 불리언 배열

따라서 불리언 배열을 numpy.sum 함수에 인자로 전달하면, 배열에서 True인 원소의 개수가 반환됩니다. 다음 예제를 보면 쉽게 이해할 수 있습니다.

import numpy as np

x = np.array([True, False, True, True, False])
print(np.sum(x))

이 코드의 결과는 3입니다. 이는 불리언 배열에서 True인 원소의 개수와 일치합니다.

 

2) numpy.count_nonzero 함수에서 불리언 배열

불리언 배열을 다룰 때 유용하게 사용되는 함수 중 하나는 numpy.count_nonzero입니다. 

이 함수는 배열에서 0이 아닌 원소의 개수를 반환합니다. 

 

따라서 불리언 배열에서 True인 원소의 개수를 구하고자 할 때는 numpy.count_nonzero 함수를 사용할 수도 있습니다. 다음 예제를 보겠습니다. 결과적으로 numpy.sum의 경우와 같은 결과를 보입니다.

import numpy as np

x = np.array([True, False, True, True, False])
print(np.count_nonzero(x))

결과는 numpy.sum 함수와 마찬가지로 3이 나옵니다.