파이썬 공부 11 - list에 for 반복문과 if 구문 넣기

 list에 for 반복문 넣기

1, 2, 3, 4, 5, 6, 7, 8, 9, 10 이라는 숫자들로 리스트를 만드려면 보통 아래와 같이 코드를 작성할 것입니다.

numbers = []
for i in range(1, 10+1):
    numbers.append(i)

print(numbers)

위의 방식을 이용하면 3줄의 코드를 작성하여 numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 라는 리스트를 만들 수 있습니다. 하지만 파이썬에서는 단 한줄의 코드만으로 똑같은 리스트를 작성할 수 있습니다.

numbers = [n for n in range(1, 10+1)]

print(numbers)

위의 코드처럼 리스트 안에 for 반복문을 넣으면 단 한줄의 코드만으로도 numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 라는 리스트를 만들 수 있습니다.

for 반복문이 안에 들어가는 리스트의 구조는 아래와 같습니다.

variable = [<result> for <each value in the list> in <how many times will loop>]

list에 if 구문 넣기

numbers = [1, 2, 4, 523, 9, 11, 32, 6]

이번엔 위의 리스트에서 짝수만 골라서 새로운 리스트를 선언해보려고 합니다. if 구문을 이용하면 짝수만 골라낼 수 있을 것입니다. 여기서 만약 if 구문을 리스트 안에다 넣을 수 있다면 작성할 코드의 숫자를 줄일 수 있을 것입니다.

numbers = [1, 2, 4, 523, 9, 11, 32, 6]

even_numbers = [n for n in numbers if n % 2 == 0]

print(even_numbers)

if 구문이 안에 들어가는 리스트의 구조는 아래와 같습니다.

variable = [<result> for <each value in the list> in <how many times will loop> if statement]