[Algorithm] 파이썬 배열 요소 인덱스 여러 개 찾기

index()

data = [1, 1, 0, 0, 0, 1]
print(data.index(1))

# 결과: 0

index()는 가장 첫 번째 인덱스, 즉 하나의 인덱스만 반환한다.
특정 요소의 인덱스를 모두 찾으려는 경우에는 enumerate를 사용하면 된다.

enumerate()

data = [1, 1, 0, 0, 0, 1]
print([i for i, value in enumerate(data) if value == 1])

# 결과: [0, 1, 5]