Python

리스트 표현식

파이썬의 리스트는 안에 for 반복문과 if 조건문을 사용할 수 있음. 리스트 안에 식, for 반복문, if 조건문 등을 지정하여 리스트를 생성하는 것을 리스트 컴프리헨션(List Comprehension)이라 함

컴프리헨션은 능력, 이해력, 시험 등의 뜻도 있지만, 어떤 것을 잡아서 담아둔다는 뜻이 있음. 즉, 식으로 지정해서 생성된 것을 리스트로 잡아두는 것이 리스트 컴프리헨션

[식 for 변수 in 리스트]
list(식 for 변수 in 리스트)

>>> a = [i for i in range(10)]
>>> a
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> b = list(i for i in range(10))
>>> b
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

• 리스트 안에 식, for, 변수, in, 리스트 순서로 들어가 있기 때문에 뒤에서 앞으로 읽으면 간단함. range(10)으로 0부터 9까지 생성하여 변수 i에 숫자를 꺼내고, 최종적으로 i를 이용하여 리스트를 만든다는 뜻

>>> c = [i + 5 for i in range(10)]
>>> c
[5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
>>> d = [i * 2 for i in range(10)]
>>> d
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

• [i for i in range(10)]는 변수 i를 그대로 사용하지만, 위과 같이 식 부분에서 i를 다른 값과 연산하면 각 연산의 결과를 리스트로 생성 가능
• [i + 5 for i in range(10)]과 같이 i에 5를 더하면 0부터 9까지의 숫자에 5를 더한 값으로 리스트를 생성. 마찬가지로 [i * 2 for i in range(10)]는 0부터 9까지의 숫자에 2를 곱한 값으로 리스트를 생성함

Tip! 대괄호와 list() 리스트 표현식

• 리스트 표현식은 [식 for 변수 in 리스트]처럼 [ ](대괄호)로 만들 수도 있고, list(식 for 변수 in 리스트)처럼 list로 만들 수도 있음.
• 하지만 둘 중에 성능은 대괄호 방식이 더 좋음
  • 특히 list 방식은 C 언어 스타일이라 대괄호 방식이 파이썬 다운 코드.
• 따라서 리스트 표현식은 대괄호 방식을 사용하는 것이 좋음. list는 리스트 표현식을 만들 수 있다는 정도만 알아 두면 됨

리스트 표현식 - if 조건문

[식 for 변수 in 리스트 if 조건식]
list(식 for 변수 in 리스트 if 조건식)

>>> a = [i for i in range(10) if i % 2 == 0]
>>> a
[0, 2, 4, 6, 8]

• [i for i in range(10) if i % 2 == 0]는 0부터 9까지 숫자를 생성하여 2의 배수인 숫자(짝수)로만 리스트를 생성함
• 즉, for 반복문 뒤에 if 조건문을 지정하면 숫자를 생성한 뒤 if 조건문에서 특정 숫자만 뽑아내서 리스트를 생성

>>> b = [i + 5 for i in range(10) if i % 2 == 1]
>>> b
[6, 8, 10, 12, 14]

• i를 다른 값과 연산해서 리스트 만드는 것도 가능
  
  

리스트 표현식 - if 조건문 else 포함

>>> [i if i % 2 == 0 else None for i in range(5) ]
[0, None, 2, None, 4]

• 위의 예제는 [0,1,2,3,4] 중 짝수는 모두 리스트에 포함시키고 홀수일 경우에는 None을 리스트에 포함시키는 경우
• else가 추가되는 경우에는 if-else문이 i와 for 사이에 위치하여야 한다는 점에 유의!
  
  

for 반복문과 if 조건문을 여러 번 사용할 때

[식 for 변수1 in 리스트1 if 조건식1     for 변수2 in 리스트2 if 조건식2     ...     for 변수n in 리스트n if 조건식n]
 
list(식 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]

• 리스트 표현식에 for가 여러 개일 때는 뒤에서부터 앞으로 순차적 처리
• 위의 예시에서는 코드를 한 줄로 입력했지만 여러 줄로 입력해도 됨
• 이때 들여쓰기는 해도 되고 하지 않아도 되지만 가독성을 위해 들여쓰기를 해주는 것이 좋음

a = [i * j for j in range(2, 10)
           for i in range(1, 10)]