[파이썬] NaN 제외하고 unique 값 얻기

파이썬(Python)에서 unique()를 사용하면 NaN이 포함된 유일값들이 array로 반환된다.

언뜻 생각하기에 뭔가 NaN을 포함시키지 않는 옵션이 있을 것 같고, NaN이 포함된 array에서 NaN을 쉽게 제거할 수 있을 것 같지만 의외로 까다롭다.

까다롭다는 게 복잡한 코드를 써야 한다는 건 아니지만, 머리속으로 유추하기 어렵고 검색은 통해야 하는 정도.

 

방법1. 애초에 NaN을 지우고 unique() 를 사용한다.

직관적이지만 불필요한 연산이 들어간다. 기존 데이터가 무겁지 않다면 이 방법도 괜찮다.

labels = targetData.dropna().unique()

 

방법2. unique() 사용 후 NaN을 지운다.

아래와 같은 코드를 사용해야 한다.

labels = targetData.unique()
labels = labels[~pd.isnull(labels)]