[python] 파이썬 배열 내 동일항목 카운트 구하기 (numpy unique)

특정 배열에서 동일한 값들의 빈도수를 구해야 하는 경우가 있습니다. 반복문을 통해 직접 카운팅을 할 수도 있지만, numpy 모듈의 unique 함수를 이용하면 보다 간결하게 해결할 수 있습니다. 아래 코드는 배열 [1,1,1,2,2,2,5,25,1,1]에서 동일한 값의 빈도수를 구합니다.

import numpy as np

data_set = np.array([1,1,1,2,2,2,5,25,1,1])

unique, counts = np.unique(data_set, return_counts=True)
data_freq = np.asarray((unique, counts)).T
print(data_freq)

실행하면 아래와 같은 결과를 얻을 수 있습니다.

[[ 1  5]
 [ 2  3]
 [ 5  1]
 [25  1]]

끝.