Daily 파이썬 #1 : 리스트 컴프리헨션(list comprehension)

오늘은 파이썬 리스트 컴프리헨션에 대해서 알아보도록 하겠습니다. 오늘의 글은 아래의 자료를 바탕으로 작성됐습니다.

• 제대로 파이썬 (리스트 컴프리헨션) 

 

이번 글에서 알아볼 내용

• 리스트 컴프리헨션이란?
• 사용방법

 

리스트 컴프리헨션이란?

컴프리헨션(comprehension) 이라는 것이 도대체 무엇일까요? 사전을 참조하여 그 뜻을 파악해보자면 이해, 포함이란 뜻을 가지고 있습니다. 여기서 포함이라는 뜻을 메인으로 사용합니다. 그러면 무엇을 포함하느냐 바로 for 문 입니다. 이 의미에 대한 이해는 아래의 코드를 봐주세요.

 

# 컴프리헨션 적용 전

normal_list = [] 
for n in range(1, 10+1):
    normal_list.append(n)

# 컴프리헨션 적용 후

comprehension_list = [x for x in range(0,10)]

 

상단에 있는 코드는 리스트 컴프리헨션이 적용되기 전 코드입니다. 리스트에 1부터 10까지의 숫자를 넣기 위해 사용되는 코드가 굉장히 길다는 것을 파악하실 수 있습니다. 하지만 아래의 컴프리헨션이 적용된 코드를 보시면 단 1줄로 끝납니다. [] 사이에 for 문을 포함(comprehension) 했기에 가능해진 일입니다. 컴프리헨션을 사용하시면 보시다시피 훨씬 파이썬스러운 코드를 작성할 수 있습니다.

 

리스트 컴프리헨션의 사용방법

 

위의 그림을 보시면 사용방법을 쉽게 이해하실 수 있습니다. 똑같이 for문을 선언해줍니다. 그 다음 여기서 중요한데요 for문에 있는 x는 for문 앞에 있는 x와 같습니다. 즉 for문에 있는 x가 1이 되면 for문 앞에 있는 x도 1이 됩니다. 이 규칙을 이용하면 여러가지 일을 할 수 있습니다. 2의 배수로 10개의 원소를 가지고 있는 리스트를 만들어보겠습니다. 아래의 코드를 확인해주세요.

 

comp_list = [2*x for x in range(0, 10)]

# 결과값 : [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

 

정말 간단하게 리스트를 구성할 수 있습니다. 앞으로 리스트에 원소를 채워넣어야 한다면 리스트 컴프리헨션의 사용을 강력히 고려해보세요.

 

마무리하면서

오늘은 파이썬 리스트 컴프리헨션에 대해서 알아봤습니다. 컴프리헨션은 리스트 뿐만 아니라 Dictionary, Set 에서도 사용할 수 있습니다. 다음 글에서는 해당 컴프리헨션에 대해서 작성해보겠습니다. 부족한 글 읽어주셔서 감사합니다.