[Python] 리스트 컴프리헨션 (list comprehension)

hukim·2020년 9월 20일
2

Python

목록 보기
6/12

리스트 컴프리헨션 (list comprehension) 이란?

리스트 컴프리헨션이란 새로운 리스트를 만들 때 사용할 수 있는 간단한 표현식이며, 리스트와 마찬가지로 대괄호 [,]를 사용합니다.

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루프 내에서 새로운 리스트를 만드는 루프는 리스트 컴프리헨션으로 변환이 가능합니다.

리스트 컴프리헨션으로 작성한 코드는 간결하고 데이터베이스를 조회하여 리스트로 만들 때 많이 사용되므로 리스트 컴프리헨션은 새로운 배열을 만들 때 특히 유용하게 쓰입니다.

0개의 댓글