立竿见影地把你的 Python 代码提速7倍

开发 后端
在我们以前的文章中,曾经讲过计算斐波那契数列的几种方法,其中基于递归的方法是速度最慢的,例如计算第40项的值,需要36秒。

 [[322806]]

在我们以前的文章中,曾经讲过计算斐波那契数列的几种方法,其中基于递归的方法是速度最慢的,例如计算第40项的值,需要36秒。如下图所示:

 

要提高运算速度,根本办法当然是改进算法。不过算法的提高是一个长期积累加上灵机一动的过程。我们今天要讲的,是一个不费脑筋,立竿见影的方法——把 Python 代码编译成 C 语言代码。通过 C 语言的运行效率来加速计算过程。

这个过程看起来很复杂,但实际上你并不需要编写一行 C 语言代码。你需要做的只是使用一个叫做 Cython 的库把 Python 代码编译为 C 语言代码即可。

首先我们来安装 Cython,就像安装普通的第三方库一样:

 

  1. python3 -m pip install cython 

安装完成以后,我们单独写计算斐波那契数列的函数:

 

  1. def fib(n): 
  2.     if n in [1, 2]: 
  3.         return 1 
  4.     return fib(n - 1) + fib(n - 2) 

非常简单的递归写法。然后关键来了,我们要把这个文件保存为fast_fib.pyx。注意后缀是.pyx。如下图所示:

 

然后我们创建一个setup.py文件,文件内容如下:

 

  1. from setuptools import setup 
  2. from Cython.Build import cythonize 
  3.  
  4. setup(ext_modules=cythonize('fast_fib.pyx')) 

如下图所示:

 

这个文件的作用,就是调用 Cython 的cythonize函数把 Python 代码转换为 C 代码。

接下来,开始编译代码,执行如下命令:

 

  1. python3 setup.py build_ext --inplace 

我的 Python 是 Python3.7,所以运行完成以后,会生成一个fast_fib.cpython-37m-darwin.so,如果你的 Python 是3.8,这个文件名可能是fast_fib.cpython-38m-darwin.so。这个文件你可以改名字,例如改成fast_fib.so。

还有一个文件叫做fast_fib.c。不过你不用打开这个文件,因为它有3200多行。并且你甚至可以直接把它删掉。真正有用的只有这个fast_fib.cpython-38m-darwin.so文件。

你需要做的,仅仅是直接调用你的函数。我们另外创建一个文件test_fast_fib.py,内容如下:

 

  1. import time 
  2. from fast_fib import fib 
  3.  
  4. start = time.time() 
  5. result = fib(40) 
  6. end = time.time() 
  7. print(f'斐波拉契数列第40项为:{result},耗时:{end - start}秒'

运行效果如下图所示:

 

计算斐波那契数列第40项只需要5秒钟,速度妥妥变成 Python 版本的7倍。

使用 Cython,不仅可以提高程序的运行速度,还可以把你的核心代码转换为.so文件,防止别人反编译看到你的代码。

关于 Cython 的更多介绍,请阅读它的官方文档[1]

有同学可能会问,当前文件夹下面既然有fast_fib.pyx文件,为什么当我们执行from fast_fib import fib的时候,不会从这个文件里面导入 Python 版本的代码?

这是因为,import只会从后缀为.py/.pyc/.pyo/.so的文件中导入模块,不会进入.pyx文件中寻找。

责任编辑:华轩 来源: 未闻Code
相关推荐

2012-03-30 10:03:53

2023-03-24 12:39:55

CIOCEO

2017-04-26 10:08:57

台式机噪静音教程

2012-03-15 14:47:41

百度

2018-05-09 14:26:44

提速降费新方案

2020-03-09 15:00:35

搜狗AI

2019-09-19 10:23:18

编程手机微信

2022-07-04 15:44:26

人工智能数据中心

2024-01-16 07:48:55

百万订单系统

2011-11-15 11:03:48

2020-11-27 06:58:24

索引

2023-03-08 07:46:53

面试官优化结构体

2013-08-28 12:36:48

金万维

2018-03-28 14:10:10

GoPython代码

2021-09-02 18:02:24

腾讯QQ移动应用

2011-12-13 10:11:49

2010-05-27 22:30:08

桌面虚拟化回报

2018-08-13 20:48:20

混改5G中国联通

2022-08-09 09:10:31

TaichiPython

2022-07-26 07:05:50

PythonAPI语法
点赞
收藏

51CTO技术栈公众号