Toán tử Walrus := để gán và kiểm tra trong một dòng

Toán tử walrus := được giới thiệu trong Python 3.8 để cho phép bạn gán và kiểm tra trong một dòng. Hình dạng := nhìn giống răng nanh hải mã — và đó là lý do của tên gọi. Nó giúp giảm bớt code thừa khi cùng một giá trị được tái sử dụng bên trong điều kiện if hoặc while.

Cơ bản về toán tử walrus

= thông thường là một câu lệnh và không thể xuất hiện bên trong biểu thức — vì vậy bạn không thể đặt nó trong điều kiện của if hoặc while. := là một biểu thức, nên nó có thể đặt trực tiếp bên trong điều kiện.

Với biến := biểu_thức, bạn gán kết quả vào biến và dùng chính giá trị đó làm giá trị của điều kiện.

# Cách thường (hai dòng)
cart = ["táo", "bánh mì", "sữa", "trứng"]
n = len(cart)
if n >= 3:
    print(f"{n} món: miễn phí vận chuyển")

# Phiên bản walrus (một dòng: gán + kiểm tra)
if (n := len(cart)) >= 3:
    print(f"{n} món: miễn phí vận chuyển")
# Kết quả: 4 món: miễn phí vận chuyển

Bọc nó trong dấu ngoặc đơn khi đặt trong điều kiện

if n := len(cart) >= 3: không có dấu ngoặc đơn sẽ đánh giá len(cart) >= 3 trước và gán True/False vào n. Cách viết chuẩn là (n := len(cart)) >= 3 — bọc phép gán trong dấu ngoặc đơn.

Dùng toán tử walrus với if

Một mẫu phổ biến trong if là cần giá trị đã kiểm tra cả bên trong thân. Thông thường bạn phải tách "gán, rồi kiểm tra" ra hai dòng, nhưng toán tử walrus cho phép bạn viết một lần, ngay tại nơi giá trị được cần".

# Rẽ nhánh dựa trên số lượng kết quả tìm kiếm, rồi dùng nó trong thân
search_results = ["Sách A", "Sách B", "Sách C", "Sách D", "Sách E"]

if (hit_count := len(search_results)) >= 3:
    print(f"Tìm thấy {hit_count} kết quả")
else:
    print("Quá ít kết quả")
# Kết quả: Tìm thấy 5 kết quả

Dùng toán tử walrus để kiểm tra tổng tiền giỏ hàng.

① Định nghĩa cart_prices = [1200, 800, 3500, 600] (giá của mỗi món).

② Bên trong điều kiện if, viết (total := sum(cart_prices)) >= 5000. Khi tổng là 5000 trở lên, in "{total} yên: miễn phí vận chuyển"; còn lại in "{total} yên: cần thêm {5000 - total} yên để được miễn phí vận chuyển".

③ Sau khối if, gọi print(total) để xác nhận total vẫn có thể truy cập được bên ngoài.

(Phần giải thích sẽ hiện ra khi bạn chạy code đúng.)

Dùng toán tử walrus với while

Trong while, walrus tỏa sáng khi bạn muốn giá trị được lấy ra vừa làm điều kiện vừa là thứ bạn dùng bên trong thân.

# Làm sạch hàng đợi từng phần tử một từ đầu
pending_jobs = ["gửi email", "sao lưu", "triển khai"]

while pending_jobs and (job := pending_jobs.pop(0)):
    print(f"đang xử lý: {job}")
# đang xử lý: gửi email / đang xử lý: sao lưu / đang xử lý: triển khai

# Ngay khi pending_jobs trở thành rỗng, vế trái đánh giá False và pop() bị bỏ qua — vòng lặp thoát.

pending_jobs and (job := pending_jobs.pop(0)) thoát vòng lặp qua đánh giá ngắn mạch ngay khi vế trái rỗng (False). Điều đó tránh bug gọi pop() trên list rỗng và kích hoạt IndexError.

In danh sách ứng viên tìm kiếm từng người một từ đầu.

① Định nghĩa candidates = ["Nguyễn Văn Minh", "Trần Thị Linh", "Lê Hoàng Nam"].

② Dùng while candidates and (name := candidates.pop(0)): để lấy một người khỏi đầu và in "Ứng viên: {name}".

③ Sau vòng lặp, in "Đã kiểm tra xong tất cả ứng viên".

Khi nào nên dùng và khi nào không

Toán tử walrus không thêm khả năng gì ngoài việc tiết kiệm số ký tự gõ. Viết logic trên một dòng thì tiện, nhưng cũng có thể gây khó đọc, nên hãy chọn lọc khi dùng.

Trong bài này bạn đã học cách dùng toán tử walrus := để kết hợp phép gán và kiểm tra thành một biểu thức duy nhất.