为何含if分支的代码比数组访问代码执行快近一倍？

你的测试结果和预期不符，核心原因是测试场景中分支预测完全命中，且数组访问存在额外的隐性开销，具体拆解如下：

1. 分支预测的完美命中消除了分支开销 #

你测试时传入的x固定为5，每次调用with_if都会走最后一个elif x ==5分支。现代CPU的分支预测器会快速学习到这个固定模式，每次都精准预测分支走向，分支的流水线停顿开销几乎为0。甚至Python解释器或Java的JIT编译器会进一步优化，把固定分支直接简化为返回常量的操作，完全消除分支判断的开销。

2. 数组访问的隐性开销被你忽略 #

without_if函数存在两个关键额外开销：

• 列表重复创建：Python中函数内部的arr = ["Zero", ...]会在每次调用时重新创建一个列表对象，包括分配内存、初始化字符串引用，这是非常耗时的操作。而with_if中没有对象创建的开销，所有字符串都是预加载的字面量。
• 数组边界检查：Python（以及Java）的数组/列表索引访问都会强制进行边界检查，确保索引不越界。这个检查本质上是一个隐含的分支判断，且因为你的x=5刚好是列表的最后一个索引，这个分支每次都要执行，无法被优化掉。

验证与优化建议 #

优化数组测试代码，消除重复创建开销 #

把列表移到函数外部，避免每次调用重新创建：

import time

# 把列表定义在函数外，全局初始化一次
arr = ["Zero","One", "Two", "Three", "Four", "Five"]

def without_if(x):
    return arr[x]

num_iterations = 10000000

start_time = time.time()
for _ in range(num_iterations):
    result = without_if(5)
end_time = time.time()

print(f"Result: {result}")
print(f"Execution time without if: {end_time - start_time}")

修改后，数组访问的执行时间会显著下降，和with_if的差距会大幅缩小。

测试分支预测失效的场景 #

如果想验证分支预测的真实开销，可以让x随机取值（比如从0到5随机选择），此时with_if的分支预测会频繁失败，CPU流水线频繁停顿，执行时间会明显超过数组访问的版本。示例代码：

import time
import random

def with_if(x):
    if x == 1:
        return "One"
    elif x == 2:
        return "Two"
    elif x == 3:
        return "Three"
    elif x == 4:
        return "Four"
    elif x == 5:
        return "Five"
    else:
        return "Unknown"

arr = ["Zero","One", "Two", "Three", "Four", "Five"]
def without_if(x):
    return arr[x]

num_iterations = 10000000
random.seed(42)

# 测试随机x的分支版本
start_time = time.time()
for _ in range(num_iterations):
    result = with_if(random.randint(0,5))
end_time = time.time()
print(f"Execution time with random if: {end_time - start_time}")

# 测试随机x的数组版本
start_time = time.time()
for _ in range(num_iterations):
    result = without_if(random.randint(0,5))
end_time = time.time()
print(f"Execution time with random array: {end_time - start_time}")

此时你会看到分支版本的执行时间显著高于数组版本，符合你最初对分支预测开销的预期。

查看字节码/编译后的代码 #

可以用Python的dis模块查看两个函数的字节码差异，直观看到without_if中的BUILD_LIST操作：

import dis
dis.dis(without_if)
dis.dis(with_if)

without_if每次调用都会执行BUILD_LIST来创建新列表，而with_if只有常量加载和比较操作，没有额外的对象创建开销。

内容的提问来源于stack exchange，提问作者이윤수