python 使用 Dis 模块进行代码性能剖析

Python实用宝典

Python代码在执行的时候，会被编译为Python字节码，再由Python虚拟机执行Python字节码。有时候就我们执行python文件的时候会生成一个pyc文件，这个pyc文件即用于存储Python字节码指令，而这些字节码是一种类似于汇编指令的中间语言，但是每个字节码对应的不是机器指令，而是一段C代码。

而Dis模块，就是用于查看这些字节码的运行轨迹，因此我们可以用Dis模块判断两个函数的内存占用谁会更大，谁会更消耗CPU性能，不仅如此，通过指令，我们还可以知道Python中一些内置函数、变量的取值过程、运行逻辑，对于我们代码性能并优化代码很有帮助。

下面将通过两个例子，来介绍Dis模块的使用。

1.为什么下面第一个函数比第二个函数耗得内存更少？

def test1(a):
    if 0 < a and a < 1:
        return 1
    return 0

比

def test2(a):
    if 0 < a < 1:
        return 1
    return 0

一般人是比较难直接看出来的，但是我们使用Dis模块却能很容易找到答案：

import dis
def test1(a):
    if 0 < a and a < 1:
        return 1
    return 0
def test2(a):
    if 0 < a < 1:
        return 1
    return 0
dis.dis(test1)