[파이썬 문법] 리스트 컴프리헨션 List Comprehension

리스트 컴프리헨션 (List Comprehension)은 리스트를 초기화하는 방법 중 하나다. 

대괄호 안에 조건문과 반복문을 이용해 간결하게 리스트를 생성할 수 있다. 

array = [i for i in range(15)]

print(array)

# out:
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]

2차원 리스트를 초기화할 때 사용하면 편리하다. 

2021.05.24 - [Language/Python] - [파이썬 문법] 리스트 출력

 

아래의 코드로 numbers의 원소 중 홀수만 추릴 수 있다. 

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
odd_numbers = []

for number in numbers:
  if number % 2 == 1:
    odd_numbers.append(number)

print(odd_numbers)

# out:
[1, 3, 5, 7, 9]

아래의 리스트 컴프리헨션을 적용한 코드는 위 코드와 같은 기능을 한다. 

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
odd_numbers = [number for number in numbers if number % 2 == 1]

print(odd_numbers)

# out:
[1, 3, 5, 7, 9]