内包表記内の複数の箇所で同じ計算結果を参照したい、ということはよくあることです。


例えば、文房具を扱う会社の受注プログラムを作成するとします。
注文がきた時点で、発送可能な出荷単位以上の在庫があるかどうかを確認し、顧客に表示できるようにしましょう。





このロジックを辞書内包表記で記述します。





この内包表記をみてバッと目に付くのは、get_units(name) が繰り返し記述されていることです。
これによってコードの簡潔さが失われちゃってますし、目にウルサいコードになってしまっています。
このように、内包表記内で同じ計算結果を参照したい、でも不要な繰り返しはしたくない、という場合の最も簡単な対処方法は、Python 3.8 で採用された walrus operator (:=) を利用することです。





assignment expression を利用したおかげで、値の取得、変数へのセット、条件判定が１箇所で済んだばかりではなく、取得結果を使い回すことでコード全体がスッキリしました。
しかも、関数の呼び出し回数が減ったことによるパフォーマンスの向上も期待できます。


さて、assignment expression は内包表記内の値を設定するための書式中に記載することもできます。





しかし上の例では内包表記の評価順序を無視して変数を参照しようとしたためにエラーになっています。
つまり、for 文 -> if 文 -> 値の設定、という順序ですね。
if 文で tenth 変数を参照しようとしてもこの変数が定義されるのは最後の「値の設定」段階ですから、「まだ存在していませんよ」と怒られちゃうわけです。
このエラーを解決するには、assignment expression を if 文で記述し、そこで定義した変数を「値の設定」段階で参照するようにします。





assignment expression はもちろんジェネレータ書式内でも同様に利用できます。
辞書インスタンスではなく、商品名と在庫単位を返すイテレータを作成すると次のようになります。