walrus(ウォルラス)演算子:=入門: 使い方と実用例(Python 3.8+)

text = "This is a sample text."
# 従来の書き方(代入と利用が別々)
n = len(text)
if n > 10:
    print(n)

# walrus(代入と条件判定を同時に)
if (n := len(text)) > 10:
    print(n)

# バージョン確認と、walrus利用可否の表示
import sys

print("Python version:", sys.version)
usable = sys.version_info >= (3, 8)
print("walrus ':=' usable?", usable)

# 文字列の長さを一度だけ計算し、条件と本文の両方で使う
text = "hello walrus"
if (n := len(text)) > 10:  # len(text) を n に代入し、同時に n > 10 を判定
    print(f"長い文字列です: 長さ={n}")
else:
    print(f"短い文字列です: 長さ={n}")

# StringIOでファイル風のオブジェクトを作成し、空行で終了する例
from io import StringIO

f = StringIO("apple\nbanana\n\ncherry\n")  # 空行で打ち切るイメージ
count = 0

while (line := f.readline().strip()):  # 1行読みつつ空行でループ終了
    print(f"読み取り: {line}")
    count += 1

print(f"合計 {count} 行を処理しました")

# チャンク読み(空文字列でEOF)の例
from io import StringIO

f = StringIO("abcdefghijklmnopqrstuvwxyz")
size = 5
chunks = []

while (chunk := f.read(size)):  # 読み込んだchunkが空なら終了
    chunks.append(chunk)

print(chunks)

# 文字列リストから、strip後に数字ならint化して2倍にする
raw = [" 10", "x", "  7", "42 ", "3.14", "  -5"]

# m := s.strip() を条件と変換の両方で再利用
doubles = [n * 2
           for s in raw
           if (m := s.strip()).isdigit() and (n := int(m)) >= 0]

print(doubles)  # 'x' や '3.14' や '-5' は除外される

# 高コストな計算を模した関数
def expensive(x: int) -> int:
    print(f"expensive({x}) を実行")
    return x * x

x = 10

# walrusなし(2回呼ばれるので非効率)
# if expensive(x) > 50:
#     print(expensive(x))

# walrusあり(1回だけ呼ぶ)
if (res := expensive(x)) > 50:
    print(f"結果: {res}")
else:
    print("小さいです")

s = "abcdef"

# 良い例: 括弧で意図が明確
if (n := len(s)) > 3:
    print(n)

# 悪い例: 括弧がないと読み手が優先順位を誤解しやすい
# if n := len(s) > 3:  # これは (len(s) > 3) を n に代入してしまうので意図とズレる
#     ...

# ユーザーに何度も入力を促し、数字なら合計に加算、空行で終了する
total = 0
print("数字を入力してください(空行で終了):")

while (s := input("> ").strip()):
    if s.isdigit():
        total += int(s)
        print(f"OK: 合計={total}")
    else:
        print("数字ではありません。もう一度。")

print(f"最終合計: {total}")

import re

text = "Name: Alice, Age: 30"
pat = re.compile(r"Name:\s*(?P<name>\w+),\s*Age:\s*(?P<age>\d+)")

if (m := pat.search(text)):  # マッチオブジェクトを m に束縛
    print(f"名前={m.group('name')}, 年齢={m.group('age')}")
else:
    print("マッチしませんでした")

# 空行で処理終了、先頭が '#' の行(コメント)は飛ばす
from io import StringIO

f = StringIO("# comment\nalpha\nbeta\n\nomega\n")

while (line := f.readline()):
    line = line.rstrip("\n")
    if not line:  # 空行で終了
        break
    if line.startswith("#"):
        continue
    print(f"処理: {line}")

# バイナリ風のチャンク読み(EOFで終了)
from io import BytesIO

g = BytesIO(b"0123456789")
bufs = []
while (buf := g.read(3)):  # 空bytes(b'')で終了
    bufs.append(buf)

print(bufs)

# 正整数文字列だけを抽出し、int化したうえで平方にする
raw = ["10", "x", "003", "-5", "7"]

squares = [n * n
           for s in raw
           if (t := s.strip()).isdigit() and (n := int(t)) >= 0]

print(squares)

# 典型的な間違い例(コメントで示します)

# NG: if n = 5:   # 代入文は式に使えないためSyntaxError
# OK: if n == 5:

# OK: if (m := re.search(pat, s)):  # 代入式は式として使える
#     ...

# 内包表記のスコープ確認
vals = [1, 2, 3]
res = [(y := v * 2) for v in vals]  # y は外側スコープにも現れる
print(res)

print(y)  # 最後に束縛された 6 が残る

[2, 4, 6]
6

# OK
if (m := len("abc")) > 0:
    pass

# NG例(コメント)：以下はSyntaxErrorになるため使えません
# (obj.attr := 1)            # 属性への代入は不可
# (d["key"] := 2)            # 添字への代入は不可
# ((a, b) := (1, 2))         # 複数代入(アンパック)は不可

# バージョンに応じて書き分ける例
import sys

def compute_len(s: str) -> int:
    return len(s)

s = "walrus"

if sys.version_info >= (3, 8):
    if (n := compute_len(s)) > 3:
        print(f"3.8+ OK: {n}")
else:
    n = compute_len(s)
    if n > 3:
        print(f"fallback: {n}")