6.2.4 for 문 한 줄로 작성하기

파이썬에서는 아주 간결하게 한 줄로 for 문을 작성할 수도 있습니다. 형식은 다음과 같습니다.

[동작 for 변수 in 반복 대상]

형식만으로 이해하기 어려우니 예제로 살펴봅시다. 나도고등학교는 출석번호가 1, 2, 3, 4, 5, ... 순이었는데 새 학기부터는 각 번호에 100을 더한 형태, 즉 101, 102, 103, 104, 105, ... 식으로 바뀐다고 합니다. 이를 코드로 작성해 보겠습니다.

먼저 리스트에 기존 출석번호 5개를 저장합니다. 그런 다음 한 줄 for 문을 다음과 같이 작성해 값을 변경합니다.

TIP | 한 줄 for 문은 정확한 용어로 리스트 컴프리헨션(list comprehension)이라고 합니다. for뿐만 아니라 if 문의 조건도 함께 사용할 수 있지만, 이 책에서는 다루지 않습니다.

students = [1, 2, 3, 4, 5]
print(students)

# 한 줄 for 문으로 각 항목에 100 더하기
students = [i + 100 for i in students]
print(students)

[1, 2, 3, 4, 5]
[101, 102, 103, 104, 105]