[Python] Comprehension (List, Dictionary, Tuple)

🖐 List Comprehension

 

👌 리스트 생성하기

* 1~10까지 정수를 순서대로 가지고 있는 리스트를 생성하는 코드 
numbers = []
for n in range(1, 10+1):
	numbers.append(n)   

* 리스트 컴프리헨션(List Comprehension) : 리스트를 생성하는 컴프리헨션 
[x for x in range(10)]

 

 

👌 for 문과 리스트 컴프리헨션 비교

→ 공통점 : 리스트를 생성하는 방식은 대괄호([ ]) 를 통해 생성

→ 차이점 : 리스트 컴프리헨션은 리스트 내부에 코드를 작성

* 1~10까지 숫자 중 짝수만 순차적으로 들어있는 리스트 생성하기 
* for문
even_numbers = []
for n in range(1, 10+1):
	if n % 2 == 0:
    	even_numbers.append(n)
        
* 리스트 컴프리헨션
[ x for x in range(1, 10+1) if x % 2 == 0 ]

# [실행 결과] # [2, 4, 6, 8, 10]

* 중복 표현 예시

* 저녁으로 먹을 메뉴 쌈밥, 치킨, 피자 중 하나와 후식으로 먹을 사과, 아이스크림, 커피 를
  정할때 가능한 경우의 수
[ (x, y) for x in ['쌈밥', '치킨', '피자'] for y in ['사과', '아이스크림', '커피'] ]  

# [실행 결과] 
# [('쌈밥', '사과'), 
# ('쌈밥', '아이스크림'), 
# ('쌈밥', '커피'), 
# ('치킨', '사과'), 
# ('치킨', '아이스크림'), 
# ('치킨', '커피'), 
# ('피자', '사과'), 
# ('피자', '아이스크림'), 
# ('피자', '커피')]

* 위의 예시에서 배달에 대한 선택지 추가  
* for문
for x in ['쌈밥', '치킨', '피자']:
	for y in ['사과', '아이스크림', '커피']:
    	for z in ['배달 시키기', '가서 먹기']:
        	print(x, z, y)

* 리스트 컴프리헨션
[ (x, y, z) for x in ['쌈밥', '치킨', '피자'] for y in ['사과', '아이스크림', '커피']
for z in ['배달 시키기', '가서 먹기']]

# [실행 결과]
# ['배달 시키기', '가서 먹기']]
# [('쌈밥', '배달 시키기', '사과'),
#  ('쌈밥', '가서 먹기', '사과'), 
#  ('쌈밥', '배달 시키기', '아이스크림'), 
#  ('쌈밥', '가서 먹기', '아이스크림'), 
#  ('쌈밥', '배달 시키기', '커피'), 
#  ('쌈밥', '가서 먹기', '커피'), 
#  ('치킨', '배달 시키기', '사과'), 
#  ('치킨', '가서 먹기', '사과'), 
#  ('치킨', '배달 시키기', '아이스크림'), 
#  ('치킨', '가서 먹기', '아이스크림'), 
#  ('치킨', '배달 시키기', '커피'), 
#  ('치킨', '가서 먹기', '커피'), 
#  ('피자', '배달 시키기', '사과'),
#  ('피자', '가서 먹기', '사과'), 
#  ('피자', '배달 시키기', '아이스크림'), 
#  ('피자', '가서 먹기', '아이스크림'), 
#  ('피자', '배달 시키기', '커피'), 
#  ('피자', '가서 먹기', '커피')]

* 여러 조건 설정하기 (if문 중복)  

** 리스트 컴프리헨션 
[ x for x in range(10) if x < 5 if x % 2 == 0 ]

# [실행 결과] 
# [0, 2, 4]

 

🖐 Dictionary Comprehension

👌 set comprehension 만들기

→ 리스트 컴프리헨션 문법을 중괄호({ }) 를 이용해서 생성

* 리스트 컴프리헨션
[ x+y for x in range(10) for y in range(10) ]

# [실행 결과] 
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10,  
# 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 3, 4, 5, 6, 7, 8, 9, 10, 11,  
# 12, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 5, 6, 7, 8, 9, 10, 11,  
# 12, 13, 14, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 7, 8, 9, 10,  
# 11, 12, 13, 14, 15, 16, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17,  
# 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]  


* set comprehension { x+y for x in range(10) for y in range(10) } 

# [실행 결과] 
# {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18}

 

 

👌 dictionary comprehension 만들기

  → key : value 의 형태로 set comprehension 내용 채우기

* 학생들이 한달간 달린 거리를 저장할 dictionary 생성, 처음 거리는 0으로 지정  

* dictionary comprehension  
students = ['철수', '영희', '길동', '순희'] 
{ students: 0 for student in students }  

# [실행 결과] 
# {'철수': 0, '영희': 0, '길동': 0, '순희': 0}

* 전학을 간 학생의 성적 제거  

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

# [실행 결과] 
# {'길동': 90, '순희': 60, '영희': 80, '철수': 50}

* scores 에 따른 통과, 불통과 조건 지정  
grades = { name: 'PASS' if value > 70 else 'No-Pass' for name, value in scores.items() } 
print(grades)  

# [실행 결과] 
# {'길동': 'PASS', '순희': 'No-PASS', '영희': 'PASS', '철수': 'No-PASS'}

 

🖐 Tuple

👌 Tuple Comprehension

 → 컴프리헨션으로 만든 정보를 수정할 필요가 없는 정보일 때 생성

* Tuple Comprehension  ( x for x in range(10) )

# [실행 결과]
# <generator object <genexpr> at 0x104955f68>

 

📢 generator : 값을 넘겨주면 값을 소비한 후 더 이상 정보를 기억하지 않음