[BW 29] 대입식을 사용해 컴프리헨션 안에서 반복 작업을 피하라

회사에서 주문을 관리하기 위한 프로그램을 작성한다고 하자.

고객의 주문 요청이 재고 수량을 넘지 않아야 한다.

stock = {
    '못': 125,
    '나사못': 35,
    '나비너트': 8,
    '와셔': 24,
}

order = ['나사못', '나비너트', '클립']

def get_batches(count, size):
    return count // size

result = {}
for name in order:
    count = stock.get(name, 0)
    batches = get_batches(count, 8)
    if batches:
        result[name] = batches

print(result)

>>>
{'나사못': 4, '나비너트': 1}

 

위 코드를 컴프리헨션으로 바꾸면 다음과 같다.

found = {name: get_batches(stock.get(name, 0), 8)
        for name in order
        if get_batches(stock.get(name, 0), 8)}
print(found)

>>>
{'나사못': 4, '나비너트': 1}

하지만 get_batches(stock.get(name, 0), 8)이 반복된다는 단점이 있다.

 

이때, 왈러스 연산자(:=)를 사용하여 대입식을 만들면 가독성과 성능이 좋아진다.

found = {name: batches for name in order
        if (batches := get_batches(stock.get(name, 0), 8))}
print(found)

>>>
{'나사못': 4, '나비너트': 1}