[파이썬 numpy] 유용한 where 함수 (조건에 맞는 값의 위치, 값 변경)

[파이썬 numpy] 

유용한 where 함수 (조건에 맞는 값의 위치, 값 변경)

where 함수는 두가지 기능을 합니다.

- 특정 조건에 맞는 원소의 위치를 반환

- 특정 조건에 맞는 원소와 그렇지 않은 원소를 원하는 값으로 변경

특정 조건에 맞는 원소의 위치를 반환

배열을 하나 정의합시다. 

>>> A=np.array([1,2,3,4,5,6])

3보다 큰 값의 위치를 구해봅시다. 

>>> np.where(A>3)

(array([3, 4, 5], dtype=int64),)

특정 조건에 맞는 원소와 그렇지 않은 원소를 원하는 값으로 변경

위 배열에서 3보다 큰 원소는 100으로, 그렇지 않은 원소는 0으로 바꿔봅시다.

>>> np.where(A>3,100,0)

array([  0,   0,   0, 100, 100, 100])