Numpy 사용법 정리(3)

1.  Unique한 원소 찾기

np.unique()를 사용하면 array에서 unique한 원소만을 추출할 수 있다. 위의 예제에서는

1차원 array에 대해 np.unique()를 적용했다. 

아래 예제에서는 2차원 array에서 적용 예제를 살펴보겠다.

 

2차원 array에 대해서 np.unique('array이름', axis=0 또는 1)을 사용하면 2차원 array에서

unique한 행이나 열을 반환하는 것을 알 수 있다. 위의 예제를 보면 쉽게 이해할 수 있을 것이다. 

 

아래 예제에서는 np.unique()를 종합적으로 사용한 예제이다. 

위의 예제는 np.unique('array', return_index=True, return_counts=True, axis=0)을

종합적으로 사용한 것이다. 

'return_index = True' 는 말그대로 반환 될 array의 index를 표시하는 것이다. 

위 예제에서는 [1, 1, 3, 4]의 index는 '0'번, [1,1,7,8]은 '3'번이기 때문에 예제와 같이 출력되었다.

 

'return_counts = True'는 반환될 array의 unique한 원소의 갯수를 표시하는 것이다. 

위 예제에서는 [1, 1, 3, 4]는 총 2개가 존재하기 때문에 '2'가 반환된 것이다. 

 

'axis=0'은 unique한 행을 반환하라는 것이며, 만약 'axis=1'이면 열이 반환될 것이다.