파이썬 강의 노트 - 리스트 간편 생성

LIST COMPREHENSION

다른 프로그래밍 언어와 달리 파이썬에서는 리스트를 보다 간단하게 생성할 수 있는 방법이 있습니다.

원래 문법대로라면 아래처럼 만들어야 합니다.

capitalized_cities = []
for city in cities:
    capitalized_cities.append(city.title())

보다 간단하게 위의 세 줄을 아래처럼 한 줄로 줄일 수 있습니다.

capitalized_cities = [city.title() for city in cities]

조건 추가하기

 if  만 추가하는 경우는 뒤에 바로 써도 괜찮습니다.

squares = [x**2 for x in range(9) if x % 2 == 0]

 else  도 같이 추가하는 경우엔 앞으로 옮겨서 작성해야 합니다.

squares = [x**2 for x in range(9) if x % 2 == 0 else x + 3]  # 에러 발생

squares = [x**2 if x % 2 == 0 else x + 3 for x in range(9)]  # 올바른 작성법