Daily Python #3 : 컴프리헨션 규칙

이번 글은 Python 컴프리헨션의 규칙에 대해서 포스팅하겠습니다. 이 규칙을 이해하면 List, Dictionary, Set 모두에서 컴프리헨션을 사용하실 수 있습니다. 즉 파이썬스러운 코드를 작성하실 수 있는거죠. 그럼 바로 시작하도록 하겠습니다.

 

이번 글에서 알아볼 내용

• for 문
• 다중 for 문
• 조건문

 

for 문

comp_list = [x for x in range(10]

# 결과 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

 

for문은 컴프리헨션의 모든 것 입니다. for 문에서 선언된 변수는 for 문 앞에서 사용할 수 있습니다. 그리고 for 문 앞에 있는 부분은 List, Dictionary, Set 의 Value 가 됩니다. 이 사실을 이용하면 굉장히 파이썬스럽게 코드를 작성할 수 있는데요. 2의 배수를 노드로 가지고 있는 리스트를 생성해보겠습니다.

 

comp_list = [x*2 for x in range(10)]

# 결과 : [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

 

정말 쉽고 간단하게 2의 배수 리스트를 생성했습니다. 이런식으로 리스트 컴프리헨션의 for 문을 잘 사용하시면 효과적이고 아름다운 코드를 작성하실 수 있습니다.

 

중첩 for 문

컴프리헨션은 중첩 for 문 또한 제공하는데요. 왼쪽에서 오른쪽으로 중첩 for문을 구성합니다. 간단한 예제코드를 통해 이 사실을 파악해보도록 합시다.

 

comp_list = [i * j for i in range(3) for j in range (3)]

# result = [0, 0, 0, 1, 2, 3, 2, 4, 6]

 

조건문

마지막으로 조건문입니다. 조건문은 for 문의 오른쪽에 선언합니다. 그리고 조건문이 True 일때 값을 반환합니다. 이해를 위해 아래의 코드를 확인해봅시다.

 

comp_list = [x for x in range(10) if x % 2 == 0]

# result : [0, 2, 4, 6, 8]

 

 

마치면서

이상으로 컴프리헨션 규칙 설명글을 마치도록 하겠습니다. 앞으로 List, Dictionary, Set 등에 값을 넣을때는 컴프리헨션 사용을 강력 고려해보시길 바랍니다. 부족한 글 읽어주셔서 감사합니다.