Python 中最快的循环方式

somenzz

发布于 2022-10-25 20:08:07

7150

发布于 2022-10-25 20:08:07

文章被收录于专栏：Python七号

大家好，我是 somenzz，今天我们来研究一下 Python 中最快的循环方式。

各种姿势

比如说有一个简单的任务，就是从 1 累加到 1 亿，我们至少可以有 7 种方法来实现，列举如下：

1、while 循环

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

2、for 循环

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

3、sum range

def sum_range(n=100_000_000):
    return sum(range(n))

4、sum generator（生成器）

def sum_generator(n=100_000_000):
    return sum(i for i in range(n))

5、sum list comprehension（列表推导式）

def sum_list_comp(n=100_000_000):
    return sum([i for i in range(n)])

6、sum numpy

import numpy
def sum_numpy(n=100_000_000):
    return numpy.sum(numpy.arange(n, dtype=numpy.int64))

7、sum numpy python range

import numpy
def sum_numpy_python_range(n=100_000_000):
    return numpy.sum(range(n))

上述 7 种方法得到的结果是一样的，但是消耗的时间却各不相同，你可以猜测一下哪一个方法最快，然后看下面代码的执行结果：

import timeit

def main():
    l_align = 25
    print(f'{"1、while 循环":<{l_align}} {timeit.timeit(while_loop, number=1):.6f}')
    print(f"{'2、for 循环':<{l_align}}  {timeit.timeit(for_loop, number=1):.6f}")
    print(f'{"3、sum range":<{l_align}} {timeit.timeit(sum_range, number=1):.6f}')
    print(f'{"4、sum generator":<{l_align}} {timeit.timeit(sum_generator, number=1):.6f}')
    print(f'{"5、sum list comprehension":<{l_align}} {timeit.timeit(sum_list_comp, number=1):.6f}')
    print(f'{"6、sum numpy":<{l_align}} {timeit.timeit(sum_numpy, number=1):.6f}')
    print(f'{"7、sum numpy python range":<{l_align}} {timeit.timeit(sum_numpy_python_range, number=1):.6f}')

if __name__ == '__main__':
    main()

执行结果如下所示：

比较快的方式

for 比 while 块

for 和 while 本质上在做相同的事情，但是 while 是纯 Python 代码，而 for 是调用了 C 扩展来对变量进行递增和边界检查，我们知道 CPython 解释器就是 C 语言编写的，Python 代码要比 C 代码慢，而 for 循环代表 C，while 循环代表 Python，因此 for 比 while 快。