一日一技:立竿见影地把你的 Python 代码提速7倍

简介: 一日一技:立竿见影地把你的 Python 代码提速7倍

摄影:产品经理又是高嘌呤的一天

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

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

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

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

python3 -m pip install cython

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

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

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

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

from setuptools import setup
from Cython.Build import cythonize
setup(ext_modules=cythonize('fast_fib.pyx'))

如下图所示:

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

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

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,内容如下:

import time
from fast_fib import fib
start = time.time()
result = fib(40)
end = time.time()
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文件中寻找。

目录
相关文章
|
4天前
|
Python
python如何获取代码运行时间?
【6月更文挑战第3天】python如何获取代码运行时间?
16 3
|
1天前
|
人工智能 数据挖掘 Python
【Python实用技能】建议收藏:自动化实现网页内容转PDF并保存的方法探索(含代码,亲测可用)
【Python实用技能】建议收藏:自动化实现网页内容转PDF并保存的方法探索(含代码,亲测可用)
|
1天前
|
人工智能 数据挖掘 Python
【AI大模型应用开发】【LangChain系列】实战案例1:用LangChain写Python代码并执行来生成答案
【AI大模型应用开发】【LangChain系列】实战案例1:用LangChain写Python代码并执行来生成答案
|
1天前
|
机器学习/深度学习 人工智能 算法
【AI大模型应用开发】【补充知识】文本向量化与向量相似度(含Python代码)
【AI大模型应用开发】【补充知识】文本向量化与向量相似度(含Python代码)
|
1天前
|
Linux 网络安全 开发工具
【超详细!超多图!】【代码管理】Python微信公众号开发(3)- 服务器代码上传Github
【超详细!超多图!】【代码管理】Python微信公众号开发(3)- 服务器代码上传Github
|
1天前
|
缓存 监控 程序员
Python装饰器:一种优雅的代码增强技术
Python装饰器是无损函数或类修改功能的工具,常用于日志、性能监控、权限验证、缓存和异常处理。它们本质是可调用对象,接收函数并返回新函数。【6月更文挑战第5天】
14 0
|
1天前
|
存储 算法 Java
必会的10个经典算法题(附解析答案代码Java/C/Python看这一篇就够)(二)
必会的10个经典算法题(附解析答案代码Java/C/Python看这一篇就够)(二)
8 1
|
1天前
|
存储 算法 Java
必会的10个经典算法题(附解析答案代码Java/C/Python看这一篇就够)(一)
必会的10个经典算法题(附解析答案代码Java/C/Python看这一篇就够)(一)
6 0
|
4天前
|
并行计算 异构计算 Python
python代码torch.device("cuda:0" if torch.cuda.is_available() else "cpu")是什么意思?
【6月更文挑战第3天】python代码torch.device("cuda:0" if torch.cuda.is_available() else "cpu")是什么意思?
22 4
|
5天前
|
Python 缓存 开发者
Python中的装饰器:优雅而强大的代码增强工具
在Python编程中,装饰器是一种强大的工具,它允许开发者在不改变原有代码结构的情况下,增加额外的功能。本文将深入探讨Python中装饰器的概念、用法和实际应用,展示其在代码优化和增强方面的威力。