为何Python使用ROT_TWO栈顶双元素交换实现变量互换？

问题背景 #

给定Python代码：

a = 1
b = 2
a, b = b, a

编译后生成的Python字节码如下：

1           0 LOAD_CONST               0 (1)
              2 STORE_NAME               0 (a)

  2           4 LOAD_CONST               1 (2)
              6 STORE_NAME               1 (b)

  3           8 LOAD_NAME                1 (b)
             10 LOAD_NAME                0 (a)
             12 ROT_TWO
             14 STORE_NAME               0 (a)
             16 STORE_NAME               1 (b)
             18 LOAD_CONST               2 (None)
             20 RETURN_VALUE

其中ROT_TWO指令用于交换栈顶前两个元素的顺序。

有人提出疑问：实现a和b的互换，似乎可以省去ROT_TWO，直接按如下字节码执行：

1           0 LOAD_CONST               0 (1)
              2 STORE_NAME               0 (a)

  2           4 LOAD_CONST               1 (2)
              6 STORE_NAME               1 (b)

  3           8 LOAD_NAME                1 (b)
             10 LOAD_NAME                0 (a)
             14 STORE_NAME               1 (b)
             16 STORE_NAME               0 (a)
             18 LOAD_CONST               2 (None)
             20 RETURN_VALUE

那么为何需要额外的ROT_TWO操作？该实现相比这个方案有何优势？

解答 #

其实这是Python编译器为了统一处理多变量赋值逻辑而做的设计，并非只针对两个变量的特殊场景：

• 本质上，a, b = b, a属于元组解包赋值：右边的b, a会先被打包成隐式元组，编译器需要先把右边所有元素按顺序加载到栈上，再按左边变量的顺序依次弹出赋值。
  对于b, a，栈上会先压入b的值，再压入a的值，此时栈顶是a、次顶是b。但左边需要给a赋值b、给b赋值a，所以必须交换栈顶两个元素的顺序——ROT_TWO就是干这个的，这样后续执行STORE_NAME时，弹出的第一个值是b（给a），第二个是a（给b），完全匹配赋值需求。

• 你设想的方案虽然能实现两个变量交换，但只适配两个变量的特殊情况。如果是三个变量的交换（比如a, b, c = c, b, a），编译器会用ROT_THREE指令；如果是更多变量的解包赋值，会用UNPACK_SEQUENCE指令统一处理。这种统一的逻辑能让编译器的代码更简洁，不用为不同数量的变量编写差异化的赋值逻辑，降低维护成本。

• 从执行效率上看，ROT_TWO是专门的单字节指令，执行速度极快，和你省去它直接调换赋值顺序的效率几乎无差别，但统一的字节码生成规则能让整个Python解释器的逻辑更一致、更易维护。

内容的提问来源于stack exchange，提问作者Samuele B.