리스트 컴프리헨션이란 새로운 리스트를 만들 때 사용할 수 있는 간단한 표현식이며, 리스트와 마찬가지로 대괄호 [,]를 사용합니다.
for문 뒤에 if문을 추가하여 조건문을 포함한 형식도 올 수 있습니다.
[표현식 for 원소 in 반복 가능 객체] [표현식 for 원소 in 반복 가능 객체 if 문]
odd_numbers = []
for element in range(1, 11):
if(element % 2) == 1:
odd_numbers.append(element)
위 코드는 홀수 값들을 골라내서 odd_numbers라는 리스트에 추가하는 코드입니다.
이 코드를 if문을 포함한 리스트 컴프리헨션으로 코드를 수정해보겠습니다.
list_comprehension = [element for element in range(1,11) if (element % 2) == 1]
print(list_comprehension)
For루프 내에서 새로운 리스트를 만드는 루프는 리스트 컴프리헨션으로 변환이 가능합니다.
리스트 컴프리헨션으로 작성한 코드는 간결하고 데이터베이스를 조회하여 리스트로 만들 때 많이 사용되므로 리스트 컴프리헨션은 새로운 배열을 만들 때 특히 유용하게 쓰입니다.