提升Python代码运行速度的5个技巧

1. 引言

如果有人问你 “什么是最快的编程语言?”，你可能会说"肯定不是Python!”

其实,Python比我们想象的运行的要快。我们之所以有先入为主的认为Python运行慢,可能是我们平常的误用和缺乏使用技巧知识。

接下来让我们看看如何用一些简单的Trick来提高我们程序的运行性能。

2. 使用内置函数

Python中的许多内置函数都是用C实现的，并且经过了很好的优化。因此，如果熟悉这些内置函数，就可以提高Python代码的性能。一些常用的内置函数有sum()、len()、map()、max()等。

假设我们有一个包含单词的列表，我们希望每个单词的首字母均变为大写。此时使用map()函数是不错的选择。

一般版本:

new_list = []
word_list = ["i", "am", "a", "python", "programmer"]
for word in word_list:
    new_list.append(word.capitalize())
改进版本: 
word_list = ["i", "am", "a", "python", "programmer"]
new_list = list(map(str.capitalize, word_list))
时间对比: 
import time
new_list = []
word_list = ["i", "am", "a", "python", "programmer"]
start = time.time()
# Slower:
for word in word_list:
    new_list.append(word.capitalize())
print(time.time() - start, "seconds")
start = time.time()
# Faster:
new_list = list(map(str.capitalize, word_list))
print(time.time() - start, "seconds")
运行结果: 
1.0013580322265625e-05 seconds
4.76837158203125e-06 seconds
可以看出第二种方法运行速度快了将近2倍. 
3. 字符串连接 VS join()
 
在Python中，字符串是不可变的，因此我们不能修改它们。
 每次当我们连接多个字符串时，我们将会创建一个新的字符串,此时会导致一些运行性能问题。 
一般版本: 
new_list = []
word_list = ["I", "am", "a", "Python", "programmer"]
for word in word_list:
    new_list += word
改进版本: 
word_list = ["I", "am", "a", "Python", "programmer"]
new_list = "".join(word_list)
时间对比: 
import time
new_list = []
word_list = ["I", "am", "a", "Python", "programmer"]
start = time.time()
for word in word_list:
    new_list += word
print(time.time() - start, "seconds")
start = time.time()
new_list = "".join(word_list)
print(time.time() - start, "seconds")
运行结果: 
4.0531158447265625e-06 seconds
9.5367431640625e-07 seconds
使用Join()函数可以让代码运行快4倍. 
4. 创建列表和字典的方式
 
一般来说,使用[]和{}来创建列表和字典相比使用list()和dict{}运行更加高效.这是因为使用list()和dict{}来创建对象时需要调用一个附加函数. 
一般版本: 
list()
dict()
改进版本: 
时间对比:
 为了便于对比时间,这里我们使用timeit函数来统计,我们运行1百万次,来看二者的时间对比,代码如下: 
import timeit
slower_list = timeit.timeit("list()", number=10**6)
slower_dict = timeit.timeit("dict()", number=10**6)
faster_list = timeit.timeit("[]", number=10**6)
faster_dict = timeit.timeit("{}", number=10**6)
print(slower_list, "seconds")
print(slower_dict, "seconds")
print(faster_list, "seconds")
print(faster_dict, "seconds")
运行结果: 
0.08825178800000001 seconds
0.083323732 seconds
0.019935448999999994 seconds
0.027835573000000002 seconds
可以看出,我们的运行速度快了将近4倍. 
5. 使用 f-Strings
 
我们已经知道将字符串进行串联可能会使程序变慢。
 另一个比较好的解决方案是使用f-Strings。 
一般版本: 
me = "Python"
string = "Make " + me + " faster"
改进版本: 
me = "Python"
string = f"Make {me} faster"
时间对比: 
import time
me = "Python"
start = time.time()
string = "Make " + me + " faster"
print(time.time() - start, "seconds")
start = time.time()
string = f"Make {me} faster"
print(time.time() - start, "seconds")
运行结果: 
2.1457672119140625e-06 seconds
9.5367431640625e-07 seconds
可以看出,我们的运行速度快了将近2倍. 
6. 使用Comprehensions
 
Python中的List Comprehensions为我们提供了更短的语法，甚至只有一行代码来实现各种强大的功能。很多用到循环的场景下,我们尽量使用生成式的语法来实现. 
一般版本: 
new_list = []
existing_list = range(1000000)
for i in existing_list:
    if i % 2 == 1:
        new_list.append(i)
较快版本: 
existing_list = range(1000000)
new_list = [i for i in existing_list if i % 2 == 1]
时间对比: 
import time
new_list = []
existing_list = range(1000000)
start = time.time()
for i in existing_list:
    if i % 2 == 1:
        new_list.append(i)
print(time.time() - start, "seconds")
start = time.time()
new_list = [i for i in existing_list if i % 2 == 1]
print(time.time() - start, "seconds")
运行结果: 
0.16418218612670898 seconds
0.07834219932556152 seconds
可以看出,我们的运行速度快了将近2倍. 
7. 附录- Python中的内置函数
 
我们可以通过官网来查看Python的内置函数. 

 如果我们只关注上述例子中一些短小的代码片段,这些技巧似乎没有太大的改善。 实际上，我们的项目很容易变得复杂,此时上述技巧就派上用场啦！ 
8. 总结
 
本文重点介绍了在Python中如何使用一些简单的Trick来提升代码运行效率，并给出了相应的代码示例。 
您学废了吗？ 

 关注公众号《AI算法之道》，获取更多AI算法资讯。