파이썬 enumerate 내장함수 사용하기

내 마음대로 파이썬 - map 내장함수 사용하기

알고리즘 문제를 풀면서 자주 사용하는 파이썬 내장함수에 대해 살펴보고 있습니다.

이전에 map 함수에 대해 정리한 바가 있는데요, 이번 정리에서는 파이썬의 내장함수인 enumerate 함수에 대해 알아보겠습니다.

 

enumerate(iterable, start)

enumerate 함수는 첫 번째 인자로 iterable 객체를 전달 받습니다.

두 번째 인자를 입력하면 시작 인덱스 값을 지정할 수 있습니다.

디폴트 시작 인덱스 값은 0입니다.

함수를 실행하면 2가지 값에 대한 접근이 가능합니다.

바로 iterable 객체의 각 원소와 원소별 인덱스 입니다.

 

students = ['mason', 'harriet', 'john']
for idx, student in enumerate(students):
    print(idx, student)
    # 0 mason
    # 1 harriet
    # 2 john

enumerate 함수는 기본적으로 인덱스와 원소로 구성된 튜플을 생성합니다.

 

students = ['mason', 'harriet', 'john']
for idx, student in enumerate(students, start=100):
    print(idx, student)
    # 100 mason
    # 101 harriet
    # 102 john

start 매개변수의 값을 별도로 전달하면 해당 값을 시작 인덱스로 지정합니다.

 

students = ['mason', 'harriet', 'john']

students_dict = {student: idx for idx, student in enumerate(students)}
print(students_dict)  # {'mason': 0, 'harriet': 1, 'john': 2}

enumerate 함수를 사용해서 리스트의 정보를 딕셔너리로 옮길 수 있습니다.

딕셔너리 컴프리헨션을 사용하는 방법인데 알고리즘 문제를 풀면서 요긴하게 사용 가능합니다.