嫌Python太慢了?让Cython带你飙车吧
本文转载自公众号“读芯术”(ID:AI_Discovery)
很多人仅在性能方面将Python与C进行了比较,没有在快速开发方面比较,就开始抱怨Python太慢了。Python的优点在于,其开发时间非常短,你不必处理指针、内存管理等问题。
多数情况下,开发时间比性能更重要。Python有一个很棒的社区,其中有很多很好的库,很多需求在那都能得到满足。
话虽如此,如果想让Python代码更快该怎么做呢?来试试Cython吧!
Cython: Python的C扩展
如果你同时喜欢Python的简单语法以及C的高性能,那么Cython就是你想要的。你可以使用Cython为Python编写C扩展,其代码将转换为优化的C / C ++代码。它可以提供高性能,且能在Python项目中使用。
安装Cython
这需要C编译器,你可以根据自己的操作系统安装一个C编译器,然后需要使用如下命令安装Cython:
pipinstall Cython
从静态类型声明中获益
可以将静态类型声明添加到变量和函数,它将提供更好的性能。你可以将所有C类型如int,float,double等用于声明。
在Cython中声明变量
cdef int i =10
这句命令的基本意思是声明一个变量i,它是一个整数。
在Cython中声明函数
cdef intsquare(int x): return x ** 2
该命令基本上意味着square函数采用整数x作为参数并返回整数值x ** 2,即x的平方。导入模块时,cdef声明的函数对Python代码不可见。为了使其可见需要改用cpdef。
在Python中计算阶乘
编写一个Python函数来计算给定整数的阶乘:
effactorial(n): if n >=1: return n *factorial(n -1) return1
使用Cython让Python更快
首先,要创建一个pyx文件。
cpdef longfastfactorial:此函数将返回一个long值,因此我们通过在函数名fastfactorial之前放置long来声明它的类型为long。
long n:我们预期n为long值。
cpdef long fastfactorial(long n): if n >=1: return n * fastfactorial(n-1) return1
接着,我们需要创建一个setup.py文件来将Cython转换为C。
from distutils.core import setup fromCython.Buildimport cythonize setup(ext_modules =cythonize('fastfac.pyx'))
编译代码
为执行编译,需要运行以下命令:
python3setup.py build_ext –inplace
--inplace:此选项会在你的工作目录中创建一个共享对象文件。
对比
让我们比较一下二者:
from fastfac import fastfactorial from fac import factorial from timeit import timeit print(timeit('fastfactorial(20)', globalsglobals=globals(), number=10000)) #output: 0.002164322999306023 print(timeit('factorial(20)', globalsglobals=globals(), number=10000)) #output: 0.18900858898996376
结果惊人,Cython快将近88倍!