[python] 넘파이 배열에서 어떤 값의 위치를 알고 싶다면, np.where 함수

넘파이 배열에서 어떤 값이 존재하는 위치를 알고 싶으면 where 함수를 사용하면 됩니다. 이름에서 알 수 있듯이 위치를 알려주는 함수입니다. 

 

 

위 예제에서 볼 수 있듯이 사용법은 간단합니다.

 

np.where(넘파이배열==찾을값)

 

을 입력하시면, 인덱스값을 반환해줍니다. 'grape'의 경우 3번째에 위치해있기 때문에 인덱스값 2가 반환되었습니다. 'apple'은 인덱스값 0, 'banana'는 1, 'grape'는 2, 'watermelon'은 3이기 때문입니다. 파이썬의 경우 인덱스값이 0부터 시작합니다. 

 

 

 

2차원 배열에서 특정 값의 위치 찾기

이번에는 2차원 배열에서 어떤 값의 위치를 찾아보겠습니다. 

 

 

위와 같이 생긴 넘파이 배열에서 4가 어디 있는지 찾아봤더니 [1][0]에 있다고 알려줬습니다. 즉, 배열에서 두번째 행, 첫번째 열에 4가 존재한다고 알려준 것입니다. 

 

관련 글

[python] numpy 패키지 유용한 함수들 10개 정리  

[python] numpy 다차원배열 클래스 객체 유용한 메소드 20개 정리  

 

참고자료

[1] numpy.org/doc/stable/reference/generated/numpy.where.html, NumPy