리스트 컴프리헨션

소개

기본 구문

doubled = [x*2 for x in range(5)]
print(doubled)

조건 사용

evens = [x for x in range(10) if x % 2 == 0]
print(evens)

문자열 변환

words = ['hello', 'world']
print([w.upper() for w in words])

중첩 컴프리헨션

matrix = [[1,2,3],[4,5,6]]
flat = [x for row in matrix for x in row]
print(flat)

표현식의 if/else

lst = [-1, 2, -3, 4]
clipped = [x if x > 0 else 0 for x in lst]
print(clipped)

컴프리헨션과 map/filter 비교

nums = range(10)
result = [x**2 for x in nums if x%2==0]
print(result)

컴프리헨션으로 평탄화하기

nested = [[1,2],[3,4],[5]]
flat = [x for sub in nested for x in sub]
print(flat)

enumerate()와 컴프리헨션

colors = ['red', 'green', 'blue']
indexed = [(i, c) for i, c in enumerate(colors)]
print(indexed)

zip()과 컴프리헨션

a = [1, 2, 3]
b = [10, 20, 30]
print([x+y for x,y in zip(a,b)])

컴프리헨션을 사용하지 않을 때(NOT)

# Too complex for comprehension:
result = []
for x in range(10):
    if x > 5:
        result.append(x * x)
print(result)

빠른 확인

복습

계속 학습하기