Python中return语句在if/else分支不同位置的执行逻辑问题

原始参考代码 #

def func(a) -> Optional[S]:
    if len(a) == 2:
         x = "bla bla bla"
    else:
        if a.state is True and len(a) == 3:
             s = "hello world"
    return s

注：s是S类型的实例。

问题1：原始代码中return s的执行时机 #

当return s处于和最外层if/else块同级的位置（即外层else块正下方、函数末尾）时：

• 执行时机：该语句是函数的全局收尾执行语句，只要代码运行过程中没有提前抛出异常、没有提前触发其他return逻辑，无论走哪个分支，最终都会执行这条return语句。
• 不存在“仅内层if条件不成立才触发return”的逻辑：
  • 当len(a) == 2时，代码直接进入最外层if块完成x赋值，根本不会进入外层else块、也不会执行内层if判断，走完x赋值后会直接跳到函数末尾执行return s，这个触发路径和内层if条件没有任何关系。
  • 当len(a) != 2进入外层else块时，无论内层if的两个判断条件（a.state is True、len(a) == 3）是否成立，走完内层if的判断逻辑后，最终也会跳到函数末尾执行return s。

注意：原始代码存在明显的变量作用域隐患：如果执行路径没有给s赋值（比如走len(a)==2分支，或者进入else但内层if条件不满足），执行return s时会直接抛出NameError，因为变量s未被定义。

问题2：return位置调整后的逻辑差异 #

调整后代码如下，return s被移动到外层else块内部、与内层if同级：

def func(a) -> Optional[S]:
   if len(a) == 2:
      x = "bla bla bla"
   else:
       if a.state is True and len(a) == 3:
          s = "hello world"
       return s

和原始写法相比，核心逻辑差异有三点：

• 执行覆盖范围不同：调整后的return s仅在进入外层else块（即len(a) != 2）时才会执行。如果len(a) == 2，代码完成x赋值后就会结束函数，不会执行任何显式return语句，函数默认返回None，完全不会触发对s变量的查找。
• 报错触发条件不同：原始代码只要走len(a)==2分支，执行return时就会因为s未定义抛出NameError；调整后的代码只有进入else块、且内层if条件不满足（未执行s = "hello world"赋值）时，才会在执行return时触发NameError，len(a)==2分支不会触发该错误。
• 返回值逻辑不同：原始代码所有能正常运行到末尾的路径，都会尝试返回s变量；调整后的代码len(a)==2分支固定返回None，仅len(a)!=2的分支会尝试返回s。

内容的提问来源于stack exchange，提问作者HelloWorld