리스트에서 0이 아닌 곳의 위치 찾기

오늘은 리스트에서 원소들 중 0이 아닌 곳의 위치를 인덱스로 반환하는 numpy의 nonzero 함수에 대해 알아보겠다. 사용법은 아래와 같다. [3,3,2,2,0] 에서 0이 아닌 곳의 위치의 인덱스인 0,1,2,3 을 반환한 것을 확인할 수 있다.

여기서 유의해야 할 점은 np.nunzero() 의 결과에 [0]과 같이 인덱싱을 해주어야한다는 점이다. 여기서 인덱싱을 해주지 않으면 ()와 같이 무언가로 감싼 결과로 결과물이 출력되어 우리가 이 인덱스를 바로 사용하기가 어렵게 되기 때문이다....