Python List, Set, Dict Comprehension

JS · 11 Oct 2020

Comprehension은 Collection을 초기화할 때 for, if를 사용하여 초기화 하는 것을 말합니다.

Python3에서는 List, Set, Dict에 대한 Comprehension을 지원합니다.

Comprehension을 사용하는 방법과 예제들을 소개합니다.

List Comprehension

다음은 Comprehension을 사용하지 않고 for문으로 List를 초기화하는 코드입니다.

squares = []
for i in range(1, 10):
    squares.append(i * i)

print(squares)

Output:

[1, 4, 9, 16, 25, 36, 49, 64, 81]

Comprehension을 사용하면 다음과 같이 간단하게 초기화할 수 있습니다. 결과는 위의 코드와 동일합니다. [ ] 안에 반복문이 있습니다.

squares = [i * i for i in range(1, 10)]
print(squares)

Output:

[1, 4, 9, 16, 25, 36, 49, 64, 81]

Syntax는 다음과 같습니다.

[expression for item in list]

for, if를 사용한 Comprehension

for와 if를 동시에 사용한 Comprehension 예제입니다. 짝수만 리스트에 할당합니다.

even_numbers = [ x for x in range(20) if x % 2 == 0]
print(even_numbers)

Output:

[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

Set Comprehension

Set Comprehension은 List와 거의 동일합니다. 대신 Set의 특성상, Set에 저장되는 아이템의 순서는 보장되지 않습니다.

squares = {i * i for i in range(1, 10)}
print(squares)

Output:

{64, 1, 4, 36, 9, 16, 49, 81, 25}
even_numbers = { x for x in range(20) if x % 2 == 0}
print(even_numbers)

Output:

{0, 2, 4, 6, 8, 10, 12, 14, 16, 18}

Dictionary Comprehension

다음은 Dictionary Comprehension을 사용하지 않고 for문으로 초기화하는 코드입니다.

squares = dict()
for i in range(1, 10):
    squares[i] = i * i
print(squares)

Output:

{1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}

Dict comprehension을 이용한다면 다음과 같이 구현할 수 있습니다. 생성되는 Dict는 위의 결과와 동일합니다.

squares = {i: i*i for i in range(1, 10)}
print(squares)

Output:

{1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}

Syntax는 다음과 같습니다.

dictionary = {key: value for vars in iterable}

예제

다음은 dollor를 pound로 변형하는 예제입니다.

다음과 같이 이전에 생성된 Dict를 for loop에 넣어 새로운 Dict를 생성할 수 있습니다.

old_price = {'milk': 1.02, 'coffee': 2.5, 'bread': 2.5}

dollar_to_pound = 0.76
new_price = {item: value*dollar_to_pound for (item, value) in old_price.items()}
print(new_price)

Output:

{'milk': 0.7752, 'coffee': 1.9, 'bread': 1.9}

다음은 Dict에서 value가 짝수인 아이템만 필터링하는 예제입니다. 여기서는 for와 함께 if를 사용하고 있습니다.

original_dict = {'jack': 38, 'michael': 48, 'guido': 57, 'john': 33}

even_dict = {k: v for (k, v) in original_dict.items() if v % 2 == 0}
print(even_dict)

Output:

{'jack': 38, 'michael': 48}

참고

댓글을 보거나 쓰려면 이 버튼을 눌러주세요.
codechachaCopyright ©2019 codechacha