[BW 10] 대입식을 사용해 반복을 피하라

대입식은 := 를 사용하고 왈러스 연산자라고도 부른다.

만약 기존의 코드가 아래와 같을 때, count 변수는 if 문의 첫 번째 블록 안에서만 사용된다.

count = fresh_fruit.get('레몬', 0)
if count:
    print("레몬 있어요! " + str(count) + "개")
else:
    print("없어요!")

이를 다음과 같이 왈러스 연산자를 사용해서 수정할 수 있다.

if count := fresh_fruit.get('레몬', 0):
    print("레몬 있어요! " + str(count) + "개")
else:
    print("없어요!")

만약 다음과 같은 while 문이 있다고 해도 왈러스 연산자를 사용하면 불필요한 변수 선언을 줄이고 가독성 있게 수정할 수 있다.

# 수정 전
bottles = []
fresh_fruit = pick_fruit()
while fresh_fruit:
    for fruit, count in fresh_fruit.items():
        batch = make_juice(fruit, count)
        bottles.extend(batch)
    fresh_fruit = pick_fruit()

# 수정 후
bottles = []
while fresh_fruit := pick_fruit():
    for fruit, count in fresh_fruit.items():
        batch = make_juice(fruit, count)
        bottles.extend(batch)