Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

파이썬에서 시퀀스를 생성하는 방법 – 리스트 컴프리헨션 활용 가이드

파이썬(Python)에서 시퀀스를 생성해야 할 때 가장 먼저 떠올려야 할 도구가 바로 리스트 컴프리헨션(List Comprehension)입니다. 리스트 컴프리헨션은 반복문과 조건문을 한 줄로 압축하여 표현할 수 있는 강력한 문법으로, 복잡한 시퀀스도 간결하고 효율적으로 만들어낼 수 있습니다.

기본 사용 예제

예를 들어, 0부터 시작하는 처음 100개의 정수를 담은 리스트가 필요하다면 다음과 같이 작성할 수 있습니다.

a = [i for i in range(100)]
print(a)

실행 결과

위 코드를 실행하면 아래와 같은 출력을 얻을 수 있습니다.

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,
24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65,
66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87,
88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99]

조건문을 활용한 응용 예제

리스트 컴프리헨션의 진짜 힘은 조건문과 결합할 때 드러납니다. 예를 들어, 처음 10개의 짝수에 대한 제곱값을 구하고 싶다면 if 조건을 추가하기만 하면 됩니다.

a = [i * i for i in range(20) if i % 2 == 0]
print(a)

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

[0, 4, 16, 36, 64, 100, 144, 196, 256, 324]

정리

리스트 컴프리헨션은 기본 반복문보다 코드가 짧고 가독성이 높으며, 내부적으로 최적화되어 있어 실행 속도 면에서도 유리한 경우가 많습니다. 이 문법에 익숙해지면 조건 필터링, 중첩 반복, 함수 적용 등 훨씬 더 강력하고 다양한 방식으로 시퀀스를 생성할 수 있습니다. 파이썬 코드를 더 깔끔하게 작성하고 싶다면 리스트 컴프리헨션 활용법을 꼭 익혀두시기 바랍니다.