在本文中,我将介绍一些简单的方法,可以将Python for循环的速度提高1.3到900倍。
Python内建的一个常用功能是timeit模块。下面几节中我们将使用它来度量循环的当前性能和改进后的性能。
对于每种方法,我们通过运行测试来建立基线,该测试包括在10次测试运行中运行被测函数100K次(循环),然后计算每个循环的平均时间(以纳秒为单位,ns)。
几个简单方法
1、列表推导式
结果如下:
可以看到使用列表推导式可以得到2倍速的提高
2、在外部计算长度
如果需要依靠列表的长度进行迭代,请在for循环之外进行计算。
通过将列表长度计算移出for循环,加速1.6倍,这个方法可能很少有人知道吧。
3、使用Set
在使用for循环进行比较的情况下使用set。
在使用嵌套for循环进行比较的情况下,使用set加速498x
4、跳过不相关的迭代
避免冗余计算,即跳过不相关的迭代。
这个方法要在设计for循环内容的时候进行代码设计,具体能提升多少可能根据实际情况不同:
5、代码合并
在某些情况下,直接将简单函数的代码合并到循环中可以提高代码的紧凑性和执行速度。
这样也可以提高1.3倍
这是为什么呢?
调用函数涉及开销,例如在堆栈上推入和弹出变量、函数查找和参数传递。当一个简单的函数在循环中被重复调用时,函数调用的开销会增加并影响性能。所以将函数的代码直接内联到循环中可以消除这种开销,从而可能显著提高速度。
⚠️但是这里需要注意,平衡代码可读性和函数调用的频率是一个要考虑的问题。
一些小技巧
6.避免重复
考虑避免重复计算,其中一些计算可能是多余的,并且会减慢代码的速度。相反,在适用的情况下考虑预计算。
结果如下
7、使用Generators
生成器支持延迟求值,也就是说,只有当你向它请求下一个值时,里面的表达式才会被求值,动态处理数据有助于减少内存使用并提高性能。尤其是大型数据集中
可以看到提升很明显:
8、map()函数
使用Python内置的map()函数。它允许在不使用显式for循环的情况下处理和转换可迭代对象中的所有项。
使用Python内置的map()函数代替显式的for循环加速了970x。
这是为什么呢?
map()函数是用C语言编写的,并且经过了高度优化,因此它的内部隐含循环比常规的Python for循环要高效得多。因此速度加快了,或者可以说Python还是太慢,哈。
9、使用Memoization
记忆优化算法的思想是缓存(或“记忆”)昂贵的函数调用的结果,并在出现相同的输入时返回它们。它可以减少冗余计算,加快程序速度。
首先是低效的版本。
然后我们使用Python的内置functools的lru_cache函数。
结果如下:
使用Python的内置functools的lru_cache函数使用Memoization加速57x。
lru_cache函数是如何实现的?
“LRU”是“Least Recently Used”的缩写。lru_cache是一个装饰器,可以应用于函数以启用memoization。它将最近函数调用的结果存储在缓存中,当再次出现相同的输入时,可以提供缓存的结果,从而节省了计算时间。lru_cache函数,当作为装饰器应用时,允许一个可选的maxsize参数,maxsize参数决定了缓存的最大大小(即,它为多少个不同的输入值存储结果)。如果maxsize参数设置为None,则禁用LRU特性,缓存可以不受约束地增长,这会消耗很多的内存。这是最简单的空间换时间的优化方法。
10、向量化
向量化一般用于机器学习的数据处理库numpy和pandas
11、避免创建中间列表
使用filterfalse可以避免创建中间列表。它有助于使用更少的内存。
使用Python的内置itertools的filterfalse函数实现相同功能的改进版本。
这个方法根据用例的不同,执行速度可能没有显著提高,但通过避免创建中间列表可以降低内存使用。我们这里获得了131倍的提高
12、高效连接字符串
任何使用+操作符的字符串连接操作都会很慢,并且会消耗更多内存。使用join代替。
该测试需要一种简单的方法来生成一个较大的字符串列表,所以写了一个简单的辅助函数来生成运行测试所需的字符串列表。
结果如下:
使用连接函数而不是使用+运算符加速1.5倍。为什么连接函数更快?
使用+操作符的字符串连接操作的时间复杂度为O(n²),而使用join函数的字符串连接操作的时间复杂度为O(n)。
总结
本文介绍了一些简单的方法,将Python for循环的提升了1.3到970x。
- 使用Python内置的map()函数代替显式的for循环加速970x
- 使用set代替嵌套的for循环加速498x[技巧#3]
- 使用itertools的filterfalse函数加速131x
- 使用lru_cache函数使用Memoization加速57x