[Python|파이썬] 기본 문법 - 컴프리헨션

참고 : https://wikidocs.net/22805

 

1. 리스트 컴프리헨션(List Comprehension)

- 새로운 리스트를 간편하게 생성하는 컴프리헨션 (리스트 내부에서 코드 작성)

- 리스트 내에서 어떤 조건에 해당하는 데이터만 추출 / 값을 변경하여 새로운 리스트를 만들 때 사용

- 리스트 명 = [반복되는/활용되는 변수 for 변수 in 반복대상 if 조건]

old_list = [1,2,3,4,5,6]
new_list = [x + 1 for x in old_list if x > 3]

name_list = ['Kim', 'Choi', 'Kwak', 'Sook']
name_lower = [name.lower() for name in name_list]
name_somi = [name+'somi' for name in name_list]
name_k = [name+'kwak' for name in name_list if name.endswith('k')]

 

2. 리스트 컴프리헨션의 확장

- set comprehension : 중괄호 이용

- dictionary comprehension : 중괄호 이용 + key:value의 형태 사용

my_list = [ x+y for x in range(10) for y in range(10) ]
my_set = { x+y for x in range(10) for y in range(10) }

students = ['철수', '영희', '길동', '순희']
# 학생들의 한달간 달린 거리를 저장할 dictionary생성, 처음 거리를 0 으로 지정
students_dict = { student: 0 for student in students }

scores = {'철수': 50, '영희': 80, '길동': 90, '순희': 60, '전학생': 100}
# 전학간 학생 성적 제거
scores = { name: score for name, score in scores.items() if name != '전학생'}