[Python/Numpy] 배열에서 고유한 요소 찾을땐 unique() 함수

 

---

넘파이(Numpy)의 unique 함수는 배열에서 모든 고유한 요소를 찾고, 이들을 정렬된 형태로 반환합니다. 이 함수는 중복된 값을 제거하고 각 고유한 값이 배열에서 최초로 등장하는 위치, 빈도 등 추가적인 정보도 제공할 수 있습니다.

 

1. 기본 사용법

---

import numpy as np

array = np.array([1, 2, 2, 3, 3, 3, 4])
unique_values = np.unique(array)
print(unique_values)  # [1, 2, 3, 4]

 

2. 추가적인 반환 값

---

• return_index=True: 고유 값이 처음 등장하는 인덱스를 반환합니다.
• return_inverse=True: 원래 배열을 재구성할 때 사용할 수 있는 인덱스 배열을 반환합니다.
• return_counts=True: 각 고유 값의 등장 횟수를 반환합니다.

 

예시: 추가적인 옵션 사용

array = np.array([1, 2, 2, 3, 3, 3, 4])
unique_values, indices, inverse_indices, counts = np.unique(array, return_index=True, return_inverse=True, return_counts=True)

print("Unique values:", unique_values)  # 고유 값
print("First indices:", indices)        # 고유 값의 첫 등장 위치
print("Inverse indices:", inverse_indices)  # 원 배열 재구성용 인덱스
print("Counts:", counts)                # 각 값의 등장 횟수

 

 

---