장고에서 쿼리셋을 꺼내기 위해 반복문을 쓸 일이 많은데, 데이터 종류가 많아 코드 가독성은 떨어지다보니 list comprehension을 써야할 것 같아서 정리해보는 블로그.

사실 python 레플릿을 하다가, 문제를 엄청 쉽게 푸는 방법(링크의 1.3부터)을 요거로 찾아내서 알게 된 개념이긴 하다. 그 이후로 딱히 사용을 안 했기 때문에 좀 더 친해져서 많이 쓰기 위해 가져와봤다.

코딩도장:리스트 표현식 사용하기의 내용을 기반으로 공부했다 🤓

파이썬의 리스트가 특이한 점은 리스트 안에 for 반복문과 if 조건문을 사용할 수 있다는 점입니다. 이렇게 리스트 안에 식, for 반복문, if 조건문 등을 지정하여 리스트를 생성하는 것을 리스트 컴프리헨션(list comprehension)이라고 합니다.

comprehension은 능력, 이해력, 시험 등의 뜻도 있지만, 어떤 것을 잡아서 담아둔다는 뜻이 있습니다. 즉, 식으로 지정해서 생성된 것을 리스트로 잡아두는 것이 리스트 컴프리헨션입니다. 개념적으로 보면 "리스트 표현식"이라고 할 수 있으니 이 책에서는 간단하게 리스트 표현식이라고 쓰겠습니다.

리스트 표현식은 다음과 같이 리스트 안에 식, for 반복문을 지정한다.

반복문: [식 for 변수 in 리스트]

• list(식 for 변수 in 리스트)의 형식으로도 쓸 수 있지만, python style은 대괄호 안에 넣는 것이다. list로 묶어주는 것은 data type을 지정하는 것이기 때문에 C언어 스타일에 가깝다.

# 0부터 9까지 숫자를 생성하여 리스트 생성
>>> a = [i for i in range(10)]        
>>> a
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

# 0부터 9까지 숫자를 생성하면서 값에 5를 더하여 리스트 생성
>>> c = [i + 5 for i in range(10)]
>>> c
[5, 6, 7, 8, 9, 10, 11, 12, 13, 14]

조건문: [식 for 변수 in 리스트 if 조건식]

# 0~9 숫자 중 2의 배수인 숫자(짝수)로 리스트 생성
>>> a = [i for i in range(10) if i % 2 == 0]   
>>> a
[0, 2, 4, 6, 8]

# 0~9 숫자 중 홀수에 5를 더하여 리스트 생성
>>> b = [i + 5 for i in range(10) if i % 2 == 1]    
>>> b
[6, 8, 10, 12, 14]

반복문, 조건문 여러번 사용하기

[식 for 변수1 in 리스트1 if 조건식1 for 변수2 in 리스트2 if 조건식2 ... for 변수n in 리스트n if 조건식n]

>>> a = [i * j for j in range(2, 10) for i in range(1, 10)]
>>> a
[2, 4, 6, 8, 10, 12, 14, 16, 18, 3, 6, 9, 12, 15, 18, 21, 24, 27, 4, 8, 12, 16, 20, 24, 28, 32, 36, 5, 10, 15, 20, 25, 30, 35, 40, 45, 6, 12, 18, 24, 30, 36, 42, 48, 54, 7, 14, 21, 28, 35, 42, 49, 56, 63, 8, 16, 24, 32, 40, 48, 56, 64, 72, 9, 18, 27, 36, 45, 54, 63, 72, 81]

# 가독성을 위해 들여쓰기 
a = [i * j for j in range(2, 10)
           for i in range(1, 10)]

내 코드에 적용

롤렉스 코드 작성 시에 만들었던 모든 반복문을 가독성 및 코드 축약을 위해 list comprehension으로 바꾸어보자 😤😤😤