Python - for문에서 index 사용하기, 3가지 방법

for 반복문에서 요소와 함께 index를 함께 사용하는 방법을 소개합니다.

1. for문에 index 변수를 직접 추가

for문과 별개로 직접 index 변수를 추가하는 방법이 있습니다. index는 for 루프 안에서 1씩 증가시켜줘야 합니다. 다음에 소개되는 방법을 이용하면 더 간단한 코드로 index에 접근할 수 있습니다.

i = 0
colors = ["red", "green", "blue"]

for color in colors:
    print(f"index: {i}, element: {color}")
    i = i + 1

Output:

index: 0, element: red
index: 1, element: green
index: 2, element: blue

2. range()를 이용하여 index로 요소 접근

range()는 범위 정보를 갖고 있는 객체입니다. 아래와 같이 range를 사용하면 for문에서 index 참조가 가능하며 index로 요소에 접근할 수 있습니다.

colors = ["red", "green", "blue"]

for i in range(len(colors)):
    print(f"index: {i}, element: {colors[i]}")

Output:

index: 0, element: red
index: 1, element: green
index: 2, element: blue

참고로, range에 대해서 조금 더 설명드리면, range(3)은 range(0, 3)이 되며 [0, 1, 2]처럼 0을 포함하고 3을 포함하지 않는 Collection을 리턴합니다.

아래와 같은 코드를 수행하면 0~2까지 출력됩니다.

for i in range(0, 3):
    print(i)

0
1
2

3. enumerate()를 이용하여 요소와 index를 접근

enumerate()를 이용하면 요소와 index를 함께 접근할 수 있습니다. index의 시작 숫자는 start=로 설정할 수 있습니다.

colors = ["red", "green", "blue"]

for i, color in enumerate(colors, start=0):
    print(f"index: {i}, element: {color}")

Output:

index: 0, element: red
index: 1, element: green
index: 2, element: blue

enumerate()는 인자로 전달되는 Collection을 index와 함께 Tuple 형태로 리턴합니다.