[파이썬/Python]컴프리헨션(Comprehension)

파이썬은 컴프리헨션이라는 문법을 통해 list, dictionary 등의 작성을 간단화 할 수 있다.

 

1. 리스트 생성

# 일반 방법
nums = []
for i in range(10):
    nums.append(i)
# 컴프리헨션을 통한 방법
nums = [i for i in range(10)]

 

2. 조건문 추가( if )

# 일반 방법
nums = []
for i in range(1,11):
    if n % 2 == 0:
        nums.append(i)
 
# 컴프리헨션 방법
nums = [i for i in range(1,11) if i % 2 == 0]

 

3. 조건문 추가( if & else )

# 일반 방법
nums = []
for i in range(1,11):
    if n % 2 == 0:
        nums.append(i)
    else:
        nums.append(-1)
 
# 컴프리헨션 방법
nums = [i if n % 2 == 0 else for i in range(1,11)]

 

4. 딕셔너리 생성

기존의 대괄호( [] )를 중괄호( {} )로 변경해준다.

# 컴프리헨션 방법
{ i for i in range(10) }