Comprehension은 Collection을 초기화할 때 for, if를 사용하여 초기화 하는 것을 말합니다.
Python3에서는 List, Set, Dict에 대한 Comprehension을 지원합니다.
Comprehension을 사용하는 방법과 예제들을 소개합니다.
1. 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]
2. 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}
3. 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}
참고
Related Posts
- Python - Yaml 파일 파싱하는 방법
- Python - 파일 내용 삭제
- Python - for문에서 리스트 순회 중 요소 값 제거
- Python - 두 리스트에서 공통 요소 값 찾기
- Python - 문자열 앞(뒤)에 0으로 채우기
- Python - 공백으로 문자열 분리
- Python - 중첩 리스트 평탄화(1차원 리스트 변환)
- Python - 16진수 문자열을 Int로 변환
- Python - 두 날짜, 시간 비교
- Python f-string으로 변수 이름, 값 쉽게 출력 (변수명 = )
- Python - nonlocal과 global 사용 방법
- Python 바다코끼리 연산자 := 알아보기
- Python - pip와 requirements.txt로 패키지 관리
- Python - 딕셔너리 보기 좋게 출력 (pprint)
- Python - Requests 사용 방법 (GET/POST/PUT/PATCH/DELETE)
- Python - 온라인 컴파일러 사이트 추천
- Python - os.walk()를 사용하여 디렉토리, 파일 탐색
- Python - 문자열 비교 방법
- Python - Text 파일 읽고 쓰는 방법 (read, write, append)
- Python - 리스트에서 첫번째, 마지막 요소 가져오는 방법
- Python - 두개의 리스트 하나로 합치기
- Python - 리스트의 마지막 요소 제거
- Python - 리스트의 첫번째 요소 제거
- Python 소수점 버림, 4가지 방법
- Python 코드 안에서 버전 확인 방법
- Python 소수점 반올림, round() 예제
- Python - 리스트 평균 구하기, 3가지 방법
- Python - bytes를 String으로 변환하는 방법
- Python - String을 bytes로 변환하는 방법
- Python 버전 확인 방법 (터미널, cmd 명령어)
- Python - 람다(Lambda) 함수 사용 방법
- Python - dict 정렬 (Key, Value로 sorting)
- Python - range() 사용 방법 및 예제
- Python - 리스트를 문자열로 변환
- Python - 문자를 숫자로 변환 (String to Integer, Float)