python循环性能pk

python循环性能pk

在任何一种编程语言中，循环都是非常消耗时间的操作。假设任意一种简单的单步操作耗费的时间是1个单位，将此操作重复执行上万次，最终耗费的时间也将增长上万倍。众所周知，python不是一种执行效率较高的语言，因此，我们对于python语言里的循环性能更加关注。

for和while

while和for是python中常用的两种实现循环的方法，但它们的运行效率实际上是有差距的。使用下面的测试代码：

import timeit

def while_loop(n=100):
    i = 0
    s = 0
    while i < n:
        s += i
        i +=1
    return s


def for_loop(n=100):
    s = 0
    for i in range(n):
        s += i
    return s


def main():
    print('while loop:', timeit.timeit(while_loop, number=100))
    print('for loop:', timeit.timeit(for_loop, number=100))


if __name__ == '__main__':
    main()

运行结果：

# while loop: 0.11076588099240325
# for loop: 0.03892302500025835

可以看到：一个简单的求和的循环，for循环要比while循环快很多。

是什么导致这样的差距？

首先要说明的一点是：C代码效率优于python代码，这已经是常识。同时，Python 底层的解释器和内置函数是用 C 语言实现的。

实际上，在每次循环中，while会比for多执行两步操作：边界检查和变量i的自增。即：while i < n和i += 1，并且这两步都是显示的纯python脚本。相比之下，for循环不需要这两步额外的依赖于python脚本的操作，因此效率优于while循环，并且随着循环次数不断增加，二者的运行效率差距也越来越大。

为了证实以上的解释，我们做如下的试验：

import timeit


def while_loop(n=100):
    i = 0
    s = 0
    while i < n:
        s += i
        i += 1