欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-03-09 23:05:39
目录1. 选择合适的数据结构2. 善用强大的内置函数和第三方库3. 少用循环4. 避免循环重复计算5. 少用内存、少用全局变量总结官方原文,代码均可运行python 是世界上使用最广泛的编程语言之一。...

官方原文,代码均可运行

python 是世界上使用最广泛的编程语言之一。它是一种解释型高级通用编程语言,具有广泛的用途,几乎可以将其用于所有事物。其以简单的语法、优雅的代码和丰富的第三方库而闻名。python除了有很多优点外,但在速度上还有一个非常大的缺点。

虽然python代码运行缓慢,但可以通过下面分享的5个小技巧提升python运行速度!

首先,定义一个计时函数timeshow,通过简单的装饰,可以打印指定函数的运行时间。

这个函数在下面的例子中会被多次使用。

def timeshow(func):
    from time import time
    def newfunc(*arg, **kw):
        t1 = time()
        res = func(*arg, **kw)
        t2 = time()
        print(f"{func.__name__: >10} : {t2-t1:.6f} sec")
        return res
    return newfunc
@timeshow
def test_it():
    print("hello pytip")
test_it()

1. 选择合适的数据结构

使用正确的数据结构对python脚本的运行时间有显着影响。python 有四种内置的数据结构:

  • 列表: list
  • 元组: tuple
  • 集合: set
  • 字典: dictionary

但是,大多数开发人员在所有情况下都使用列表。这是不正确的做法,应该根据任务使用合适数据结构。

运行下面的代码,可以看到元组执行简单检索操作的速度比列表快。其中dis模块反汇编了一个函数的字节码,这有利于查看列表和元组之间的区别。

import dis
def a():
    data = [1, 2, 3, 4, 5,6,7,8,9,10]
    x =data[5]
    return x
def b():
    data = (1, 2, 3, 4, 5,6,7,8,9,10)
    x =data[5]
    return x
print("-----:使用列表的机器码:------")
dis.dis(a)
print("-----:使用元组的机器码:------")
dis.dis(b)

运行输出:

-----:使用列表的机器码:------
3 0 load_const 1 (1)
2 load_const 2 (2)
4 load_const 3 (3)
6 load_const 4 (4)
8 load_const 5 (5)
10 load_const 6 (6)
12 load_const 7 (7)
14 load_const 8 (8)
16 load_const 9 (9)
18 load_const 10 (10)
20 build_list 10
22 store_fast 0 (data)
4 24 load_fast 0 (data)
26 load_const 5 (5)
28 binary_subscr
30 store_fast 1 (x)
5 32 load_fast 1 (x)
34 return_value
-----:使用元组的机器码:------
7 0 load_const 1 ((1, 2, 3, 4, 5, 6, 7, 8, 9, 10))
2 store_fast 0 (data)
8 4 load_fast 0 (data)
6 load_const 2 (5)
8 binary_subscr
10 store_fast 1 (x)
9 12 load_fast 1 (x)
14 return_value

看下列表的机器码,冗长而多余!

2. 善用强大的内置函数和第三方库

如果你正在使用python并且仍在自己编写一些通用函数(比如加法、减法),那么是在侮辱python。 python有大量的库和内置函数来帮助你不用编写这些函数。 如果研究下,那么你会惊奇地发现几乎90%的问题已经有第三方包或内置函数来解决。

可以通过访问官方文档查看所有内置函数。你也可以在wiki python上找到更多使用内置函数的场景。

比如,现在我们想合并列表中的所有单词为一个句子,比较法自己编写和调用库函数的区别:

# ❌ 正常人能想到的方法
@timeshow
def f1(list):
    s =""
    for substring in list:
        s += substring
    return s
# ✅ pythonic 的方法
@timeshow
def f2(list):
    s = "".join(list)
    return s
l = ["i", "love", "python"] * 1000 # 为了看到差异,我们把这个列表放大了
f1(l)
f2(l)

运行输出:

f1 : 0.000227 sec
f2 : 0.000031 sec

3. 少用循环

  • 用 列表推导式 代替循环
  • 用 迭代器 代替循环
  • 用 filter() 代替循环
  • 减少循环次数,精确控制,不浪费cpu
## 返回n以内的可以被7整除的所有数字。
# ❌ 正常人能想到的方法:
@timeshow
def f_loop(n): 
    l=[]
    for i in range(n):
        if i % 7 ==0:
            l.append(i)
    return l
#  ✅ 列表推导式
@timeshow
def f_list(n):
    l = [i for i in range(n) if i % 7 == 0]
    return l
# ✅  迭代器
@timeshow
def f_iter(n):
    l = (i for i in range(n) if i % 7 == 0)
    return l
# ✅ 过滤器 
@timeshow
def f_filter(n):
    l = filter(lambda x: x % 7 == 0, range(n))
    return l
# ✅ 精确控制循环次数 
@timeshow
def f_mind(n):
    l = (i*7 for i in range(n//7))
    return l
n = 1_000_000
f_loop(n)
f_list(n)
f_iter(n)
f_filter(n)
f_mind(n)

输出为:

f_loop : 0.083017 sec
f_list : 0.056110 sec
f_iter : 0.000015 sec
f_filter : 0.000003 sec
f_mind : 0.000002 sec

谁快谁慢,一眼便知!

filter 配合lambda大法就是屌!!!

4. 避免循环重复计算

如果你有一个迭代器,必须用它的元素做一些耗时计算,比如匹配正则表达式。你应该将正则表达式模式定义在循环之外,因为最好只编译一次模式,而不是在循环的每次迭代中一次又一次地编译它。

只要有可能,就应该尝试在循环外进行尽可能多的运算,比如将函数计算分配给局部变量,然后在函数中使用它。

# ❌ 应改避免的方式:
@timeshow
def f_more(s):
    import re
    for i in s:
        m = re.search(r'a*[a-z]?c', i)
# ✅ 更好的方式:
@timeshow
def f_less(s):
    import re
    regex = re.compile(r'a*[a-z]?c')
    for i in s:
        m = regex.search(i)
s = ["abctestabc"] * 1_000
f_more(s)
f_less(s)

输出为:

f_more : 0.001068 sec
f_less : 0.000365 sec

5. 少用内存、少用全局变量

内存占用是指程序运行时使用的内存量。为了让python代码运行得更快,应该减少程序的内存使用量,即尽量减少变量或对象的数量。

python 访问局部变量比全局变量更有效。在有必要之前,应该始终尝试忽略声明全局变量。一个在程序中定义过的全局变量会一直存在,直到整个程序编译完成,所以它一直占据着内存空间。另一方面,局部变量访问更快,且函数完成后即可回收。因此,使用多个局部变量比使用全局变量会更好。

# ❌ 应该避免的方式:
message = "line1\n"
message += "line2\n"
message += "line3\n"
# ✅ 更好的方式:
l = ["line1","line2","line3"]
message = '\n'.join(l)
# ❌ 应该避免的方式:
x = 5
y = 6 
def add():
    return x+y
add()
# ✅ 更好的方式:
def add():
    x = 5
    y = 6
    return x+y
add()

总结

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注的更多内容!