[Python🌱] 파이썬 enumerate 함수 사용법/예제

enumerate함수

 

1) 반복문을 사용할때 리스트의 순서값, 즉 인덱스의 정보가 필요한 경우가 있습니다. enumerate함수는 리스트의 원소에 순서값을 부여해주는 함수입니다. 사용 예제는 아래와 같습니다.

 

>>> item = ["First", "Second", "Third"] 
>>> data = enumerate(item) 
>>> print(data, type(data))  

<class 'enumerate'> 

 

 

2) enumerate함수는 입력으로 받은 데이터와 인덱스 값을 포함하는 enumerate 객체를 리턴해줍니다.

 

>>> item = ["First", "Second", "Third"] 
>>> for val in enumerate(item): 
... 	print(val) 

(0, 'First') 
(1, 'Second') 
(2, 'Third') 

 

✔️3) enumerate함수를 사용하여 for문을 돌려보면 리스트의 원소와 인덱스가 튜플형태로 담겨있습니다. 각각을 출력해봅시다.

 

>>> item = ["First", "Second", "Third"] 
>>> for i, val in enumerate(item): 
... 	print("{} 번쨰 값은 {}입니다".format(i, val)) 

0 번쨰 값은 First입니다 
1 번쨰 값은 Second입니다 
2 번쨰 값은 Third입니다 

위 처럼 enumerate함수는 for문과 함께 자주 사용됩니다.

 

 

# 직접 처본 코드

 

item = ["First", "Second", "Thrid"]

for i, val in enumerate(item):
    print(f"{i} 번째 값은 {val} 입니다")

 

 

 

 

 

enumerate() 함수

그럼 어떻게 해야 좀 더 파이썬답게 인덱스(index)와 원소를 동시에 접근하면서 루프를 돌릴 수가 있을까요? 🐍

바로 파이썬의 내장 함수인 enumerate()를 이용하면 되는데요. for 문의 in 뒷 부분을 enumerate() 함수로 한 번 감싸주기만 하면 됩니다.

 

>>> for entry in enumerate(['A', 'B', 'C']):
...     print(entry)
...
(0, 'A')
(1, 'B')
(2, 'C')

 

 

✔️enumerate() 함수는 기본적으로 인덱스와 원소로 이루어진 튜플(tuple)을 만들어줍니다. 따라서 인덱스와 원소를 각각 다른 변수에 할당하고 싶다면 인자 풀기(unpacking)를 해줘야 합니다.

 

>>> for i, letter in enumerate(['A', 'B', 'C']):
...     print(i, letter)
...
0 A
1 B
2 C

 

❗️ 여기서 주의해야 할 점!!

 

위에 있는 것처럼 for i, letter in enumerate(리스트) 에 있는 것처럼 꼭!!

for i, letter와 같이 i, 값과 같은 형태로 쓰도록 합니다 !!

그렇게 해야 { } 딕셔너리 안에서 내가 사용하고자 하는 i값과 letter값을 배치할 수 있습니다

 

만약 그렇게 하지 않을 시 오류가 날 수도 있으므로 주의하세요 !!

 

players = ["mumu", "soe", "poe", "kai", "mine"]
print(players)
print()

pla_dic = {key: i for i, key in enumerate(players)}
print(pla_dic)
print()

pla_dic3 = {i: key for i, key in enumerate(players)}
print(pla_dic3)
print()

 

# 결과값

['mumu', 'soe', 'poe', 'kai', 'mine']

{'mumu': 0, 'soe': 1, 'poe': 2, 'kai': 3, 'mine': 4}

{0: 'mumu', 1: 'soe', 2: 'poe', 3: 'kai', 4: 'mine'}

 

 

 

 

시작 인덱스 변경

 

루프를 돌리다보면 인덱스를 0이 아니라, 1로 시작하고 싶을 때가 있습니다. 이럴 때는 enumerate() 함수를 호출할 때 start 인자에 시작하고 싶은 숫자를 넘기면 됩니다.

 

>>> for i, letter in enumerate(['A', 'B', 'C'], start=1):
...     print(i, letter)
...
1 A
2 B
3 C

 

 

>>> for i, letter in enumerate(['A', 'B', 'C'], start=101):
...     print(i, letter)
...
101 A
102 B
103 C

 

players = ["mumu", "soe", "poe", "kai", "mine"]
print(players)
print()

dic1 = {val:i for i,val in enumerate(players, start=1)}
print(dic1)

 

 

 

enumerate() 원리

 

지금까지 for 문에서 enumerate() 함수를 사용하는 방법에 대해서 알아봤는데요. enumerate() 함수 자체가 어떻게 작동하는지 좀 더 살펴보겠습니다.

 

파이썬에서 for 문은 내부적으로 in 뒤에 오는 목록을 대상으로 계속해서 next() 함수를 호출하고 있다고 생각할 수 있습니다. 따라서, 일반 리스트를 iter() 함수에 넘겨 반복자(iterator)로 만든 후 next() 함수를 호출해보면 원소들이 차례로 얻어지는 것을 알 수 있습니다.

 

>>> iter_letters = iter(['A', 'B', 'C'])
>>> next(iter_letters)
'A'
>>> next(iter_letters)
'B'
>>> next(iter_letters)
'C'

 

 

이 번에는 enumerate() 함수를 호출한 결과를 대상으로 next() 함수를 계속해서 호출해보면, 인덱스와 원소의 쌍이 튜플(tuple)의 형태로 차례로 얻어지는 것을 알 수 있습니다.

 

>>> enumerate_letters = enumerate(['A', 'B', 'C'])
>>> next(enumerate_letters)
(0, 'A')
>>> next(enumerate_letters)
(1, 'B')
>>> next(enumerate_letters)
(2, 'C')

 

 

결국, enumerate() 함수는 인자로 넘어온 목록을 기준으로 인덱스와 원소를 차례대로 접근하게 해주는 반복자(iterator) 객체를 반환해주는 함수입니다. 이 부분은 enumerate() 함수의 반환 값을 리스트로 변환해보면 좀 더 명확하게 확인할 수 있습니다.

 

>>> list(enumerate(['A', 'B', 'C']))
[(0, 'A'), (1, 'B'), (2, 'C')]

 

 

[팁] 2차원 리스트 루프

 

이제 enumerate() 함수의 작동 원리까지 배웠으니 좀 더 고급 응용 사례를 살펴볼까요?

아래와 같은 2차원 리스트나 튜플이 담고 있는 데이터를 루프를 돌면서 접근해야한다고 가정해봅시다.

 

>>> matrix = [['A', 'B', 'C'], ['D', 'E', 'F'], ['G', 'H', 'I']]

 

 

 

이 때 일반적으로 다음과 같이 중첩 for 문 내에서 행과 열의 인덱스로 데이터를 읽도록 작성을 많이 하실 거에요.

 

>>> for r in range(len(matrix)):
...     for c in range(len(matrix[r])):
...             print(r, c, matrix[r][c])
...
0 0 A
0 1 B
0 2 C
1 0 D
1 1 E
1 2 F
2 0 G
2 1 H
2 2 I

 

 

동일한 작업을 하는 코드를 enumerate() 함수를 이용해서 재작성하면 어떨까요?

 

>>> for r, row in enumerate(matrix):
...     for c, letter in enumerate(row):
...             print(r, c, letter)
...
0 0 A
0 1 B
0 2 C
1 0 D
1 1 E
1 2 F
2 0 G
2 1 H
2 2 I

 

코드가 더 깔끔하고 읽기 편하죠?! 🤗

예제 코드에서는 단순히 데이터를 출력만 하기 때문에 큰 차이를 못 느낄 수도 있지만, 실제 프로젝트에서 좀 더 복잡한 작업을 하는 경우라면 차이가 더 크게 날 거에요. 무엇보다도 2차원 배열을 다룰 때 인덱스를 사용하면 오타를 내기 쉬운데, enumerate() 함수를 사용하면 이러한 실수를 할 확률이 현저하게 줄어듭니다.

 

 

 

ref : https://www.daleseo.com/python-enumerate/