Python中的Walrus運算符（:=）是Python 3.8引入的一個新特性，允許在表達式中同時賦值和返回值。它的核心作用是減少重復計算，提升代碼簡潔性。以下是其適用的典型場景及示例：

---

1. 在循環中避免重復計算

當循環條件需要多次調用同一個函數或計算同一個表達式時，可以使用Walrus運算符將結果賦值給變量，避免重復計算。

傳統寫法：

data = [1, 2, 3, 4, 5]
while len(data) > 0:print(data.pop())

• 問題：每次循環都會計算len(data)，雖然在這里影響不大，但在復雜計算時會浪費資源。

使用Walrus運算符：

data = [1, 2, 3, 4, 5]
while (n := len(data)) > 0:print(data.pop())

• 優點：將len(data)的值賦給n，并在條件判斷中直接使用n，避免重復計算。

---

2. 在條件判斷中同時賦值變量

當需要在條件判斷中對某個表達式的結果進行判斷和賦值時，可以簡化代碼。

傳統寫法：

# 讀取文件直到遇到空行
with open("file.txt", "r") as f:line = f.readline()while line.strip() != "":print(line)line = f.readline()

使用Walrus運算符：

with open("file.txt", "r") as f:while (line := f.readline().strip()) != "":print(line)

• 優點：將f.readline().strip()的值賦給line，并在條件判斷中直接使用，避免重復賦值。

---

3. 在列表推導式或字典推導式中簡化邏輯

當需要在推導式中復用某個計算結果時，可以結合Walrus運算符。

示例：

# 計算平方數，并過濾出偶數的平方
squares = [x*x for x in range(10) if (square := x*x) % 2 == 0]
print(squares)  # 輸出：[0, 4, 16, 36, 64]

• 優點：square變量在推導式中被賦值并復用，避免重復計算x*x。

---

4. 在條件表達式（三元運算符）中賦值

在條件表達式中，可以同時賦值并使用變量。

示例：

# 根據條件賦值并返回結果
value = 10
result = (x := value) if x > 5 else (x := 0)
print(result)  # 輸出：10
print(x)       # 輸出：10

---

5. 在函數調用中復用返回值

當函數調用的返回值需要多次使用時，可以避免重復調用。

示例：

def get_data():return {"a": 1, "b": 2}# 傳統寫法
data = get_data()
if data["a"] > 0:print(f"a: {data['a']}, b: {data['b']}")# 使用Walrus運算符
if (data := get_data())["a"] > 0:print(f"a: {data['a']}, b: {data['b']}")

---

6. 在調試或日志中快速記錄變量

在調試或日志中，可以快速將中間值賦值給變量并輸出。

示例：

# 計算并記錄中間結果
result = (x := 2 + 3) * (y := 4 + 5)
print(f"x={x}, y={y}, result={result}")  # 輸出：x=5, y=9, result=45

---

不適用的場景

雖然Walrus運算符非常強大，但以下場景應謹慎使用，以免降低代碼可讀性：

1. 簡單賦值：

   # 不推薦：
   a := 10  # 錯誤！Walrus必須在表達式中使用，不能單獨使用
   

   正確寫法：

   a = 10
   

2. 全局作用域或復雜邏輯：
   避免在全局作用域或復雜嵌套表達式中濫用，可能導致代碼難以理解。

---

總結：Walrus運算符的核心價值

• 減少重復計算：避免多次調用函數或計算復雜表達式。
• 提升代碼簡潔性：在單行代碼中完成賦值和判斷。
• 適用場景：循環條件、條件判斷、推導式、函數返回值復用等。

---

示例對比

場景	傳統寫法	Walrus寫法
循環條件	while len(data) > 0: ...	while (n := len(data)) > 0: ...
函數調用復用結果	d = get_data(); if d['a'] > 0: ...	if (d := get_data())['a'] > 0: ...
列表推導式	[x*x for x in ... if x*x %2 ==0]	[x*x for x in ... if (s:=x*x)%2 ==0]

通過合理使用Walrus運算符，可以顯著提升代碼的簡潔性和效率！