Python

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

이쥬우 2022. 11. 5. 22:38
728x90

새로운 리스트를 간편하게 만들어주는 기능

new_list = [변수 활용 for 변수 in 반복대상 if 조건

 

products = ['JOA-2020', 'JOA-2021', 'SIRO-2021', 'SIRO-2022']
recall = []  # 리콜 대상 제품 리스트
for p in products:
    if p.startswith('SIRO'):  # 제품명이 SIRO 로 시작하는가?
        recall.append(p)

print(recall)
products = ['JOA-2020', 'JOA-2021', 'SIRO-2021', 'SIRO-2022']
recall = [p for p in products if p.startswith('SIRO')]

print(recall)

위와 같이 여러 줄에 해당하는 코드를 한 줄로 바꿀 수 있다. 굿

728x90