컴프리헨션 Comprehension

컴프리헨션 Comprehension

파이썬에서 리스트, 세트, 딕셔너리 등의 컬렉션을 간단하게 생성하거나 변형하는 방법 중 하나입니다. 컴프리헨션은 반복문과 조건문을 사용하여 간결하게 컬렉션을 생성하는 기법으로, 코드를 더 간단하고 가독성 좋게 작성할 수 있도록 도와줍니다.

 

---

리스트 컴프리헨션

n = 10
result = [i for i in range(n)]  # 조건문을 코드 한 줄에 간단히 작성할 수 있다.
print(result)

>>
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

arr = [10, 20, 30, 40, 50, 60, 70, 80, 90, 100]
result = [n*2 for n in arr]
print(result)

>>
[20, 40, 60, 80, 100, 120, 140, 160, 180, 200]

result = [n for n in range(10) if n % 2 == 0]
print(result)

>>
[0, 2, 4, 6, 8]

 

중첩 반복문도 표현할 수 있다.

arr = [i*j for i in range(1,4) for j in range(1, 3)]
print(arr)

>>
[1, 2, 2, 4, 3, 6]

세트 컴프리헨션

numbers = [1, 2, 3, 4, 5, 2, 3, 4]
unique_numbers = { x for x in numbers}
print(unique_numbers)

>>
{1, 2, 3, 4, 5}

딕셔너리 컴프리헨션

names = ['apple', 'banana', 'orange']
name_lengths = {name:len(name) for name in names}
print(name_lengths)

>>
{'apple': 5, 'banana': 6, 'orange': 6}