python

[python] List Comprehension

전감자(◔◡◔) 2022. 9. 22. 22:35

간단한 반복문을 통해 조건을 만족하는 리스트를 만들 수 있음 

매우 중요하니 잘 기억해두도록 하자 

'''

변수 = [표현식 for 변수 in 리스트]

2. 리스트 + for문 결합 + 단일 if 문

변수 = [표현식 for 변수 in 리스트 if 조건식]

3. 리스트 + for문 결합 + if~else 문 (3항 연산자)
변수 = [참 if문 else 거짓 for문 ]
'''

# 1. 리스트 + for문 결합
result = [v for v in range(5)]

result=[v+1 for v in range(5)]

result=[[v] for v in range(5)]
result=[[v,v+1,v+3] for v in range(5)]

#2. 리스트 + for문 결합 + 단일 if 문

result=[v for v in range(1,11) if v%2==0]

#3. 리스트 + for문 결합 + if~else 문 (3항 연산자)
result=[0 if v % 2== 0 else 1 for v in range(1,11) ]
print(result)