第2章：控制流程：深入探究 Python 条件语句

点击上方“Python学习开发”，选择“加为星标”

第一时间关注Python技术干货！

Python 3 是一种动态、解释型的高级编程语言，以其简洁明了的语法和强大的功能而广受欢迎。上一节介绍 Python 3 中的标准输入和输出，以及它们在 CPython 实现中的工作方式。本文将进入第二章节，深入探讨Python中的条件语句。

基本概念与用法

在Python中，if、elif和else关键字用于基于条件表达式的结果来执行不同的代码块。这是最基本的决策结构，允许程序根据不同的输入做出决策。

if语句

if语句的基本语法如下：

if condition:
    # 代码块1

如果condition（条件）为真（True），则执行冒号后缩进的代码块（代码块1）。条件可以是任何表达式，其结果被评估为True或False。

elif语句

elif是else if的缩写，用于在if语句之后添加多个条件分支。它的语法如下：

if condition1:
    # 代码块1
elif condition2:
    # 代码块2

如果condition1为真，执行第一个代码块；否则，如果condition2为真，执行第二个代码块。elif可以有多个，按顺序评估，直到找到第一个为真的条件。

else语句

else语句用于当所有if和elif条件都不满足时执行代码块。它的语法如下：

if condition1:
    # 代码块1
elif condition2:
    # 代码块2
else:
    # 默认代码块

如果所有前面的条件都不为真，则执行else后面的代码块。

高级部分：字节码分析

Python的字节码是解释器执行的低级指令集。对于条件语句，相关的字节码指令包括FOR_ITER、JUMP_IF_FALSE、JUMP_IF_TRUE和POP_JUMP_IF_FALSE等。

FOR_ITER: 用于循环结构，如for和while循环。JUMP_IF_FALSE: 如果栈顶的值是假（False），则跳转到指定的地址。JUMP_IF_TRUE: 如果栈顶的值是真（True），则跳转到指定的地址。POP_JUMP_IF_FALSE: 弹出栈顶的值，如果为假，则跳转到指定的地址。

if语句的实现

当解释器遇到if语句时，它会生成一个字节码序列来执行条件检查。例如：

if condition:
    do_something()

这将转换为类似下面的字节码：

LOAD_CONST              1 (condition)
CALL_FUNCTION            0
JUMP_IF_FALSE         label1
LOAD_CONST              2 (<code object for do_something>)
CALL_FUNCTION            0

如果condition为假，控制流将跳转到label1，跳过do_something()的执行。

elif和else的实现

elif和else的实现与if类似，但会涉及到更多的跳转指令。例如：

if condition1:
    do_something1()
elif condition2:
    do_something2()
else:
    do_something3()

对应的字节码可能如下：

LOAD_CONST              1 (condition1)
CALL_FUNCTION            0
JUMP_IF_FALSE         label2
LOAD_CONST              3 (<code object for do_something1>)
CALL_FUNCTION            0
JUMP                  label3
label2:
LOAD_CONST              4 (condition2)
CALL_FUNCTION            0
JUMP_IF_FALSE         label4
LOAD_CONST              5 (<code object for do_something2>)
CALL_FUNCTION            0
JUMP                  label3
label4:
LOAD_CONST              6 (<code object for do_something3>)
CALL_FUNCTION            0
label3:

在这个例子中，如果condition1为真，执行do_something1()；如果为假，检查condition2；如果condition2为真，执行do_something2()；否则执行do_something3()。

高级部分：在Cpython中的实现方式

在CPython中负责编译的代码通常位于Python/compile.c文件中。这里面包含的逻辑会根据Python代码的语法如条件语句构建对应的抽象语法树（Abstract Syntax Tree，AST），然后将AST转换成相应的字节码。字节码是Python虚拟机执行的低级指令集。

以下是C源码中处理if语句的一个简化示例

// 这是一个简化后的实现示例，并非实际源码
static int
compiler_if(struct compiler *c, stmt_ty s)
{
    basicblock *end, *next;

    assert(s->kind == If_kind);
    end = compiler_new_block(c);
    next = compiler_new_block(c);

    /* 生成条件表达式的代码 */
    VISIT(c, expr, s->v.If.test);
    ADDOP_JREL(c, POP_JUMP_IF_FALSE, next);

    /* 生成 if 代码块的代码 */
    VISIT_SEQ(c, stmt, s->v.If.body);
    ADDOP_JREL(c, JUMP_FORWARD, end);

    /* 生成 elif 和 else 代码块的代码 */
    compiler_use_next_block(c, next);
    for (int i = 0; i < asdl_seq_LEN(s->v.If.orelse); i++)
    {
        VISIT(c, stmt, asdl_seq_GET(s->v.If.orelse, i));
    }

    compiler_use_next_block(c, end);

    return 1;
}

CPython的实际实现中，当编译器遇到if语句时，它会先评估条件表达式，然后生成一个POP_JUMP_IF_FALSE字节码指令。这个指令在运行时会检查栈顶的元素，如果为False，虚拟机将会跳至指定的字节码标签位置。对于elif和else部分，类似的逻辑适用：elif实际上是后续的if语句；else部分则是在所有if和elif条件都不满足时执行的代码块。

推荐阅读

第1章：Python基础：变量与表达式的深度解析

第1章：Python基础：标准输入和输出

第1章：Python基础-Python 3 数据类型详解：数字、字符串、布尔值