[파이썬] 반복문에서 인덱스와 요소 동시 순회 - enumerate

enumerate?

파이썬에서 for문으로 리스트와 같은 객체 순회할 때, enumerate로 요소의 순서(인덱스)와 요소를 동시에 순회할 수 있다. 

리스트 말고 튜플, 딕셔너리에도 사용가능하다.

반복문(for 문)

다음과 같은 리스트가 있다고 생각해 보자.

Arr = ['Vaundy', 'Kaze', 'Ayase']

 

파이썬에서 for문으로 순회할 때 보통 다음과 같이 한다.

for value in Arr:
    print(value)
    
Vaundy
Kaze
Ayase

 

또 리스트의 인덱스를 순회할 때는 주로 다음과 같은 방법을 사용한다.

for i in range(len(Arr)):
    print(i)
    
0
1
2

 

그럼 요소와 인덱스를 동시에 출력하고 싶을 때는 어떻게 해야 할까?라고 한다면 보통 다음과 같은 코드를 대부분(+나) 작성할 것이다.

for i in range(len(Arr)):
    print(i, Arr[i])

0 Vaundy
1 Kaze
2 Ayase

근데 이런 방법은 '파이토닉(Pythonic)'하지 않다고 한다(글 쓰면서 처음 앎).

좀 더 깔끔하게 작성하기 위해 enumerate() 함수를 활용해서 인덱스와 요소를 동시에 접근해 보자.

 

enumerate() 사용

위에서 사용한 Arr 리스트를 enumerate함수로 순회해 보자.

for value in enumerate(Arr):
    print(value)

(0, 'Vaundy')
(1, 'Kaze')
(2, 'Ayase')

위와 같이 tuple의 형태로 반환되는 걸 볼 수 있다. 따라서 인덱스와 요소를 따로 접근하려면 unpacking을 해주자.

 

for i, value in enumerate(Arr):
    print(i, value)
    
0 Vaundy
1 Kaze
2 Ayase

이렇게 하면 깔끔하게 반복문에서 인덱스와 요소를 가져올 수 있다.

 

enumerate() 활용

시작하는 index 값 변경

다음과 같이 enumerate() 함수를 호출할 때, 뒤에 인자에 시작하고 싶은 index값을 넣어주면 된다.

for i, value in enumerate(Arr, 1):
    print(i, value)

1 Vaundy
2 Kaze
3 Ayase

 

문자열 순회

my_word = "Miphas Theme"

for i, value in enumerate(my_word, 1):
    print(i, value)
    
1 M
2 i
3 p
4 h
5 a
6 s
7  
8 T
9 h
10 e
11 m
12 e

 

2차원 리스트 순회

보통 2차원 리스트를 순회하며 행과 열의 인덱스를 가져올 때 다음과 같이 한다.

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

for i in range(len(Box)):
    for j in range(len(Box[i])):
        print(i, j, Box[i][j])

0 0 A
0 1 B
0 2 C
0 3 D
1 0 E
1 1 F
2 0 G
2 1 H
2 2 I

 

이제 enumerate() 함수를 사용해 보자.

for i, row in enumerate(Box):
    for j, value in enumerate(row):
        print(i, j, value)

0 0 A
0 1 B
0 2 C
0 3 D
1 0 E
1 1 F
2 0 G
2 1 H
2 2 I

매우 파이토닉하다.

 

결론

파이썬에서 for문으로 순회할 때 enumerate() 함수를 잘 활용해서 파이토닉하게 코드를 써보자.

 

---

참고

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