본문 바로가기

Work

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<4.4)와 같은 조건식이 가능하며, numpy.array 형태로 리턴받는다.

'Work' 카테고리의 다른 글

2132년  (0) 2022.01.17
회사 품질이 나아지지 않던 이유  (0) 2021.12.20
배터리 산업 뉴스  (0) 2021.11.19
고객과 프로세스  (0) 2021.11.19
Python - shutil.make_archive()  (0) 2021.11.17