python - Python if 语句中赋值

Question

有没有比这个更简单的替代方案

res = returns_value_or_none(arg)
if res:
    do_something_with(res)

或

if returns_value_or_none(arg):
    do_something_with(returns_value_or_none(arg))

将赋值和 if 条件组合到一个语句中的一个？

Answer 1

对于 Python 3.8+，PEP 572引入赋值表达式

这允许使用符号NAME := expr为表达式中的变量赋值。它可以在 if 语句中使用，例如:

if (match := pattern.search(data)) is not None:
    # Do something with match