List Comprehensions

Python의 List Comprehensions 이해하기

개요 및 기본 구조¶

list comprehension을 사용하면 리스트를 간단하게 만들 수 있으며, 리스트를 만드는 속도가 for loop에 비해 빠르다.¹ 다만 list를 만드는 것이 아니라 연산을 해야할 때는 당연히 for loop이 더 빠르다.

list comprehension의 기본 구조는 아래와 같고, 추가 연산을 부여해줄 수도 있다.

a = [i for i in range(10)]
b = [i * 10 for i in range(10)]

print(a)
print(b)

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
[0, 10, 20, 30, 40, 50, 60, 70, 80, 90]

선형대수를 공부하면서 list comprehension을 많이 사용했는데, 자세한 내용은 여기, 여기, 여기를 참고하자.

list comprehension에 if 조건문을 부여하는 방법은 아래와 같다.

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

print(a)

[0, 2, 4, 6, 8]

list comprehension에 if else 조건문을 부여하는 방법은 아래와 같다.

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

print(a)

[0, 1, 2, 1, 4, 1, 6, 1, 8, 1]

list comprehension에서는 elif를 사용할 수 없고, 아래와 같이 else와 if를 조합해서 사용해야 한다.

a = [1, 2, 3, 4, 5]
l = ['yes' if v == 1 else 'no' if v == 2 else 'idle' for v in a]

print(l)

['yes', 'no', 'idle', 'idle', 'idle']

list comprehension안에 이중 for문을 넣는 방법은 아래와 같다. 하위의 for문이 뒤로 가는 것을 확인할 수 있다.

a = [i * j for i in [4, 5, 6] for j in [1, 2, 3]]

print(a)

[4, 8, 12, 5, 10, 15, 6, 12, 18]

list comprehension안에 list comprehension을 넣는 방법은 아래와 같다. 상위의 list comprehension이 뒤로 가는 것을 확인할 수 있다. 다차원 배열을 만들 때 많이 사용된다.

a = [[i * j for i in [4, 5, 6]] for j in [1, 2, 3]]

print(a)

[[4, 5, 6], [8, 10, 12], [12, 15, 18]]

list 뿐만 아니라 다른 자료형에도 사용 가능하다.

tuple(i for i in [1, 2, 3])

print(a)

(1, 2, 3)

a = {i * i for i in [1, 1, 2, 3, 3, 4]}

print(a)

{16, 1, 4, 9}

tmp = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e': 5}
res = {k: v * 2 for (k, v) in tmp.items()}

print(res)

{'a': 2, 'b': 4, 'c': 6, 'd': 8, 'e': 10}