首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

做到这20条,Python程序性能轻松翻倍!

1.优化算法时间复杂度

算法的时间复杂度对程序的执行效率影响最大,在Python中可以通过选择合适的数据结构来优化时间复杂度,如list和set查找某一个元素的时间复杂度分别是O(n)和O(1)。不同的场景有不同的优化方式,总得来说,一般有分治,分支界限,贪心,动态规划等思想。2. 减少冗余数据如用上三角或下三角的方式去保存一个大的对称矩阵。在0元素占大多数的矩阵里使用稀疏矩阵表示。3. 合理使用copy与deepcopy对于dict和list等数据结构的对象,直接赋值使用的是引用的方式。而有些情况下需要复制整个对象,这时可以使用copy包里的copy和deepcopy,这两个函数的不同之处在于后者是递归复制的。效率也不一样:(以下程序在ipython中运行)

importcopy

a=range(100000)

%timeit-n10copy.copy(a)# 运行10次 copy.copy(a)

%timeit-n10copy.deepcopy(a)

10loops,bestof3:1.55msperloop

10loops,bestof3:151msperloop

timeit后面的-n表示运行的次数,后两行对应的是两个timeit的输出,下同。由此可见后者慢一个数量级。4. 使用dict或set查找元素python dict和set都是使用hash表来实现(类似c++11标准库中unordered_map),查找元素的时间复杂度是O(1)

a=range(1000)

s=set(a)

d=dict((i,1)foriina)

%timeit-n10000100ind

%timeit-n10000100ins

10000loops,bestof3:43.5nsperloop

10000loops,bestof3:49.6nsperloop

dict的效率略高(占用的空间也多一些)。5. 合理使用生成器(generator)和yield

%timeit-n100a=(iforiinrange(100000))

%timeit-n100b=[iforiinrange(100000)]

100loops,bestof3:1.54msperloop

100loops,bestof3:4.56msperloop

使用()得到的是一个generator对象,所需要的内存空间与列表的大小无关,所以效率会高一些。在具体应用上,比如set(i for i in range(100000))会比set([i for i in range(100000)])快。但是对于需要循环遍历的情况:

%timeit-n10forxin(iforiinrange(100000)):pass

%timeit-n10forxin[iforiinrange(100000)]:pass

10loops,bestof3:6.51msperloop

10loops,bestof3:5.54msperloop

后者的效率反而更高,但是如果循环里有break,用generator的好处是显而易见的。yield也是用于创建generator:

defyield_func(ls):

foriinls:

yieldi+1

defnot_yield_func(ls):

return[i+1foriinls]

ls=range(1000000)

%timeit-n10foriinyield_func(ls):pass

%timeit-n10foriinnot_yield_func(ls):pass

10loops,bestof3:63.8msperloop

10loops,bestof3:62.9msperloop

对于内存不是非常大的list,可以直接返回一个list,但是可读性yield更佳(人个喜好)。python2.x内置generator功能的有xrange函数、itertools包等。6. 优化循环循环之外能做的事不要放在循环内,比如下面的优化可以快一倍:

a=range(10000)

size_a=len(a)

%timeit-n1000foriina:k=len(a)

%timeit-n1000foriina:k=size_a

1000loops,bestof3:569µsperloop

1000loops,bestof3:256µsperloop

7. 优化包含多个判断表达式的顺序

对于and,应该把满足条件少的放在前面,对于or,把满足条件多的放在前面。如:

a=range(2000)

%timeit-n100[iforiinaif10

%timeit-n100[iforiinaif1000

%timeit-n100[iforiinaifi%2==andi>1900]

%timeit-n100[iforiinaifi>1900andi%2==]

100loops,bestof3:287µsperloop

100loops,bestof3:214µsperloop

100loops,bestof3:128µsperloop

100loops,bestof3:56.1µsperloop

8. 使用join合并迭代器中的字符串

In[1]:%%timeit

...:s=''

...:foriina:

...:s+=i

...:

10000loops,bestof3:59.8µsperloop

In[2]:%%timeit

s=''.join(a)

...:

100000loops,bestof3:11.8µsperloop

join对于累加的方式,有大约5倍的提升。9. 选择合适的格式化字符方式

s1,s2='ax','bx'

%timeit-n100000'abc%s%s'%(s1,s2)

%timeit-n100000'abc'.format(s1,s2)

%timeit-n100000'abc'+s1+s2

100000loops,bestof3:183nsperloop

100000loops,bestof3:169nsperloop

100000loops,bestof3:103nsperloop

三种情况中,%的方式是最慢的,但是三者的差距并不大(都非常快)。(个人觉得%的可读性最好)10. 不借助中间变量交换两个变量的值

In[3]:%%timeit-n10000

a,b=1,2

....:c=a;a=b;b=c;

....:

10000loops,bestof3:172nsperloop

In[4]:%%timeit-n10000

a,b=1,2

a,b=b,a

....:

10000loops,bestof3:86nsperloop

使用a,b=b,a而不是c=a;a=b;b=c;来交换a,b的值,可以快1倍以上。11. 使用if is

a=range(10000)

%timeit-n100[iforiinaifi==True]

%timeit-n100[iforiinaifiisTrue]

100loops,bestof3:531µsperloop

100loops,bestof3:362µsperloop

使用 if is True 比 if == True 将近快一倍。12. 使用级联比较x < y < z

x,y,z=1,2,3

%timeit-n1000000ifx

%timeit-n1000000ifx

1000000loops,bestof3:101nsperloop

1000000loops,bestof3:121nsperloop

x 13. while 1 比 while True 更快

defwhile_1():

n=100000

while1:

n-=1

ifn

defwhile_true():

n=100000

whileTrue:

n-=1

ifn

m,n=1000000,1000000

%timeit-n100while_1()

%timeit-n100while_true()

100loops,bestof3:3.69msperloop

100loops,bestof3:5.61msperloop

while 1 比 while true快很多,原因是在python2.x中,True是一个全局变量,而非关键字。14. 使用**而不是pow

%timeit-n10000c=pow(2,20)

%timeit-n10000c=2**20

10000loops,bestof3:284nsperloop

10000loops,bestof3:16.9nsperloop

**就是快10倍以上!15. 使用 cProfile, cStringIO 和 cPickle等用c实现相同功能(分别对应profile, StringIO, pickle)的包

importcPickle

importpickle

a=range(10000)

%timeit-n100x=cPickle.dumps(a)

%timeit-n100x=pickle.dumps(a)

100loops,bestof3:1.58msperloop

100loops,bestof3:17msperloop

由c实现的包,速度快10倍以上!16. 使用最佳的反序列化方式下面比较了eval, cPickle, json方式三种对相应字符串反序列化的效率:

importjson

importcPickle

a=range(10000)

s1=str(a)

s2=cPickle.dumps(a)

s3=json.dumps(a)

%timeit-n100x=eval(s1)

%timeit-n100x=cPickle.loads(s2)

%timeit-n100x=json.loads(s3)

100loops,bestof3:16.8msperloop

100loops,bestof3:2.02msperloop

100loops,bestof3:798µsperloop

可见json比cPickle快近3倍,比eval快20多倍。17. 使用C扩展(Extension)目前主要有CPython(python最常见的实现的方式)原生API, ctypes,Cython,cffi三种方式,它们的作用是使得Python程序可以调用由C编译成的动态链接库,其特点分别是:CPython原生API: 通过引入Python.h头文件,对应的C程序中可以直接使用Python的数据结构。实现过程相对繁琐,但是有比较大的适用范围。ctypes: 通常用于封装(wrap)C程序,让纯Python程序调用动态链接库(Windows中的dll或Unix中的so文件)中的函数。如果想要在python中使用已经有C类库,使用ctypes是很好的选择,有一些基准测试下,python2+ctypes是性能最好的方式。Cython: Cython是CPython的超集,用于简化编写C扩展的过程。Cython的优点是语法简洁,可以很好地兼容numpy等包含大量C扩展的库。Cython的使得场景一般是针对项目中某个算法或过程的优化。在某些测试中,可以有几百倍的性能提升。cffi: cffi的就是ctypes在pypy(详见下文)中的实现,同进也兼容CPython。cffi提供了在python使用C类库的方式,可以直接在python代码中编写C代码,同时支持链接到已有的C类库。使用这些优化方式一般是针对已有项目性能瓶颈模块的优化,可以在少量改动原有项目的情况下大幅度地提高整个程序的运行效率。18. 并行编程因为GIL的存在,Python很难充分利用多核CPU的优势。但是,可以通过内置的模块multiprocessing实现下面几种并行模式:多进程:对于CPU密集型的程序,可以使用multiprocessing的Process,Pool等封装好的类,通过多进程的方式实现并行计算。但是因为进程中的通信成本比较大,对于进程之间需要大量数据交互的程序效率未必有大的提高。多线程:对于IO密集型的程序,multiprocessing.dummy模块使用multiprocessing的接口封装threading,使得多线程编程也变得非常轻松(比如可以使用Pool的map接口,简洁高效)。分布式:multiprocessing中的Managers类提供了可以在不同进程之共享数据的方式,可以在此基础上开发出分布式的程序。不同的业务场景可以选择其中的一种或几种的组合实现程序性能的优化。19. 终级大杀器:PyPyPyPy是用RPython(CPython的子集)实现的Python,根据官网的基准测试数据,它比CPython实现的Python要快6倍以上。快的原因是使用了Just-in-Time(JIT)编译器,即动态编译器,与静态编译器(如gcc,javac等)不同,它是利用程序运行的过程的数据进行优化。由于历史原因,目前pypy中还保留着GIL,不过正在进行的STM项目试图将PyPy变成没有GIL的Python。如果python程序中含有C扩展(非cffi的方式),JIT的优化效果会大打折扣,甚至比CPython慢(比Numpy)。所以在PyPy中最好用纯Python或使用cffi扩展。随着STM,Numpy等项目的完善,相信PyPy将会替代CPython。20. 使用性能分析工具除了上面在ipython使用到的timeit模块,还有cProfile。cProfile的使用方式也非常简单: python -m cProfile filename.py,filename.py 是要运行程序的文件名,可以在标准输出中看到每一个函数被调用的次数和运行的时间,从而找到程序的性能瓶颈,然后可以有针对性地优化。

◆◆◆◆◆

现在人工智能爆发,Python是一门脚本语言,它更适合去做人工智能这个领域,在人工智能上使用Python比其他编程语言有更大的优势。

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180910B0QXZC00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券