[pythonic code] 리스트 컴프리헨션 list comprehention

💡 리스트 컴프리헨션?

리스트 컴프리헨션이란 단순하게, 좀 더 간단한 방법으로 리스트를 만드는 방법이다.

 

0부터 9까지의 숫자가 나열된 리스트를 만든다고 할 때, 일반적인 반복문을 사용해서 리스트에 숫자를 추가하는 방법을 사용한다면 다음처럼 코드를 작성한다.

result = []
for i in range(10):
	result.append(i)

 

여기서 리스트 컴프리헨션으로 리스트를 만들면 단 한줄로 코드를 작성할 수 있다.

result = [x for x in range(10)]

 

요것이 바아로 리스트 컴프리헨션이다-!

 

---

💡 리스트 컴프리헨션을 사용하는 방법 1) IF문 (필터링)

result = [i for i in range(10) if i % 2 == 0]

앞에서 작성한 리스트 컴프리헨션 코드에 if문을 추가하여 리스트에 들어가는 값을 필터링할 수 있다.

 

result = [i if i % 2 == 0 else 10 for i in range(10)]

else문까지 사용할 거라면 이렇게 if-else문을 for문 앞에 작성해야 한다.

 

 

💡 리스트 컴프리헨션을 사용하는 방법 2) 중첩반복문

>>> lst_1 = ['a1', 'a2', 'a3']
>>> lst_2 = ['b1', 'b2', 'b3']
>>> [i + j for i in lst_1 for j in lst_2]
['a1b1', 'a1b2', 'a1b3', 'a2b1', 'a2b2', 'a2b3', 'a3b1', 'a3b2', 'a3b3']

 

 

💡 리스트 컴프리헨션을 사용하는 방법 3) 이차원 리스트 (Nested List Comprehension)

이차원 리스트는 하나의 정보를 row 단위로 저장한다. 따가서 대괄호를 하나 더 사용해야 한다.

>>> lst_1 = ['a1', 'a2', 'a3']
>>> lst_2 = ['b1', 'b2', 'b3']
>>> [ [ i + j for i in lst_1 ] for j in lst_2 ]
[['a1b1', 'a2b1', 'a3b1'], ['a1b2', 'a2b2', 'a3b2'], ['a1b3', 'a2b3', 'a3b3']]

앞서 본 중첩 반복문과는 다르게 뒤의 for문이 더 먼저 실행된다. 위의 코드에서는 j가 먼저 'b1'으로 고정되고 lst_1이 돈다.

 

[ [ row[i] for row in matrix ] for i in range(4) ]

 

 

✅ +) 리스트 컴프리션 활용

 

• 각 원소에 함수 일괄 적용

>>> vec = [-4, -2, 0, 2 ,4]
>>> [abs(x) for x in vec]
[4, 2, 0, 2, 4]

 

• 각 원소에 method 일괄 적용

>>> fruits = ['   banana', 'loganberry     ', 'passion fruit   ']
>>> [fruit.strip() for fruit in fruits]
['banana', 'loganberry', 'passion fruit']

 

• tuple 자료형을 원소로 갖는 리스트 만들기

>>> [ (x, x**2) for x in range(6) ]
[(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]

>>> [ x, x**2 for x in range(6) ]
  File "<stdin>", line 1
    [ x, x**2 for x in range(6) ]
                ^
SyntaxError: invalid syntax

 

 

• 전치 행렬 만들기

>>> matrix = [[1, 2], [3,4], [5,6], [7,8]]

>>> [ [row[i] for row in matrix] for i in range(2) ]
[[1, 3, 5, 7], [2, 4, 6, 8]]

 

+) 참고

 

IT CookBook, 데이터 과학을 위한 파이썬 프로그래밍

5. Data Structures — Python 3.10.0 documentation