[딥러닝 첫걸음] numpy의 where 함수 사용법

0 - 100 사이의 랜덤 값을 가진 1차원 배열을 생성한다.

결과값과 명확한 차이를 보여주기 위해 10을 곱해주었다.

x의 값은 아래와 같이 나온다.

x = np.random.randint(0, 100, size =100)
x = x*10
x

 

여기에서 900보다 큰 값들을 찾아 보자.

np.where()함수를 쓰게되면 조건을 만족하는 값의 인덱스를 배열로 담아준다.

loc = np.where(x>900)
loc

결과를 보면 잘 찾은 걸 볼 수 있다.

 

이 값들을 확인사살 해준다. 

print(x[loc])