본문 바로가기

where()

Python - List의 요소 위치 찾기 1-D 배열을 생성하는 방법은 흔히 두 가지, builtin의 list 객체와 numpy의 array 객체 두 형태가 많이 사용된다. 근데 두 방법으로 생성된 배열 객체는 요소 위치를 다르게 찾아야 한다. a = [1.1, 2.2, 3.3, 4.4, 5.5] ### list 객체로 생성 b = numpy.arrary([1.1, 2.2, 3.3, 4.4, 5.5]) ### numpy.array 객체로 생성 print("index of 3.3 in a:", a.index(3.3)) print("where of 3.3 in b:", numpy.where(b==3.3)) 단지 사용 메소드만이 아니라, 활용면의 차이가 큰데, 당연하게도 numpy.where()가 훨씬 유연하다. 예컨대, numpy.where(b 더보기

티스토리툴바