Work
Python - List의 요소 위치 찾기
아로미랑
2021. 12. 2. 13:22
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<4.4)와 같은 조건식이 가능하며, numpy.array 형태로 리턴받는다.