Cython学习笔记1：利用Cython加速Python运行速度

Cython学习笔记1：利用Cython加速Python运行速度 CythonCython 的核心特点：利用Cython加速Python运行速度1. Cython加速Python运行速度原理2. 不使用Cython3. 使用Cython加速（1）使用pip安装 cython 和 setuptools 库（2）安装c语言编译器（3）创建要编译为动态链接库的.py文件（4）创建setup.py（5）把.py转化为库文件（6）python程序调用库（7）annotate参数（8）定义变量的类型加速python运行速度

Cython

Cython 是一个用于将 Python 代码转换为 C 语言扩展模块的编程语言。它允许你在 Python 中编写 C 风格的代码，从而提高性能，尤其是在需要大量计算的情况下。通过将 Python 代码与 C 代码混合使用，Cython 既保留了 Python 的简洁性，又能提升程序的执行速度。

Cython 的核心特点： 提高性能：Cython 可以将 Python 代码编译成 C 语言代码，然后生成高效的 C 扩展模块。它适用于需要频繁计算的任务，尤其是数值计算和循环处理。与 C 代码交互：Cython 允许直接调用 C 函数和访问 C 数据结构，从而减少了 Python 和 C 之间的接口调用开销。简化 C 接口：与直接使用 C 语言不同，Cython 通过 Python 风格的语法简化了 C 接口的使用，开发者可以快速将 Python 代码提升到 C 代码的性能。无缝集成：Cython 代码可以与现有的 Python 代码无缝集成，开发者不需要重写整个程序，通常只需要优化性能瓶颈部分。 利用Cython加速Python运行速度

以计算圆周率 π 为例子，使用Nilakantha 级数来计算 π 值，计算公式如下：

1. Cython加速Python运行速度原理

把 python 代码转化为 c 语言代码，然后再编译成动态链接库，最后使用 python 程序调用这个库

2. 不使用Cython

创建 2 个 .py文件，fun.py 和 Nilakantha_main.py

fun.py 用于写计算的函数 Nilakantha_main.py 用于调用函数，查看结果和运行时间 fun.py代码如下

def calculate_pi(count: int) -> float: pi = 3.0 sign = 1 for i in range(2, count * 2, 2): term = sign * 4.0 / (i * (i + 1) * (i + 2)) pi += term sign *= -1 return pi

Nilakantha_main.py代码如下

import time import fun def main(): start_time = time.time() print(fun.calculate_pi(20_000_000)) end_time = time.time() print('Time:', end_time - start_time) if __name__ == '__main__': main()

运行Nilakantha_main.py文件后，结果如下：

3.141592653589787 Time: 4.7696967124938965

这段代码设置了迭代次数为2千万次，最后耗时4.76秒 下面使用cython加速

3. 使用Cython加速 （1）使用pip安装 cython 和 setuptools 库

使用如下命令进行安装，使用清华镜像源，这样速度更快

pip install cython setuptools -i pypi.tuna.tsinghua.edu /simple

安装好后再安装就会出现下面的结果

（2）安装c语言编译器

下面的c语言编译器安装一款就可以了，博主是Windows11系统，安装的是Visiual Studio 2022

GCC (GNU Compiler Collection) Clang Microsoft Visual C++ (MSVC) MinGW (Minimalist GNU for Windows) （3）创建要编译为动态链接库的.py文件

新建一个名为fun_compile.py的文件，用于将它最后编译为动态链接库，让 python 程序调用 fun_compile.py代码如下

# cython: language_level=3 def calculate_pi(count: int) -> float: pi = 3.0 sign = 1 for i in range(2, count * 2, 2): term = sign * 4.0 / (i * (i + 1) * (i + 2)) pi += term sign *= -1 return pi

# cython: language_level=3

这一行代码的作用是：指示 Cython 编译器使用 Python 3 语法的指令，告诉 Cython 如何解析和编译 Python 代码，确保代码遵循 Python 3 的语法规则，而不是 Python 2

（4）创建setup.py

setup.py 是 Python 项目中用于打包和分发代码的脚本，它包含了项目的元数据和配置，用于指导 Python 工具（setuptools）如何构建、安装和分发项目。

setup.py中的代码如下

from Cython.Build import cythonize from setuptools import setup setup( ext_modules=cythonize(["fun_compile.py"]), )

fun_compile.py是编译为动态链接库的 python 代码

（5）把.py转化为库文件

当前目录下的文件如下展示

在此目录下，执行命令

python setup.py build_ext --inplace

过程展示如下

运行完成后，目录里会多出一个 .c 文件和一个库文件

（6）python程序调用库

修改Nilakantha_main.py，导入fun_compile，如果既存在fun_compile.py，也存在fun_compile.py产生的库，会优先调用动态链接库。代码如下：

import time import fun import fun_compile def main(): start_time = time.time() print(fun.calculate_pi(20_000_000)) end_time = time.time() print('Time:', end_time - start_time) start_time = time.time() print(fun_compile.calculate_pi(20_000_000)) end_time = time.time() print('Time:', end_time - start_time) if __name__ == '__main__': main()

运行结果如下： 可以看到确实加速了

（7）annotate参数

修改setup.py，加入一个参数annotate=True

from Cython.Build import cythonize from setuptools import setup setup( ext_modules=cythonize(["fun_compile.py"], annotate=True), )

然后执行命令

python setup.py build_ext --inplace

结果会多出来一个.html文件 annotate=True 参数的作用是：生成 Cython 编译时的注释文件，用于查看 Cython 将 Python 代码转换为 C 代码的详细过程和每一步的性能分析。

通过浏览器打开这个.html文件

黄色线条的代码表示翻译后的代码依赖 python 代码，颜色越深，依赖越强 点击“+”开头的行，可查看 Cython 为其生成的 c语言 代码

点开红色框的这一行，可以看到这行代码都是用 python实现的 如果想让这段代码不都用 python 来实现，可以给其中的变量添加类型，这就是这段代码没有用c语言实现的原因，因为cython不知道变量的类型。

（8）定义变量的类型加速python运行速度

修改fun_compile.py文件，代码如下：

# cython: language_level=3 import cython def calculate_pi(count: int) -> float: pi = 3.0 sign = 1 for i in range(2, count * 2, 2): term = sign * 4.0 / (i * (i + 1) * (i + 2)) pi += term sign *= -1 return pi def calculate_pi_annotated(count: int) -> float: pi: cython.double = 3.0 sign: cython.int = 1 i: cython.int for i in range(2, count * 2, 2): term: cython.double = sign * 4.0 / (i * (i + 1) * (i + 2)) pi += term sign *= -1 return pi

重新编译

python setup.py build_ext --inplace

修改Nilakantha_main.py，代码如下：

import time import fun import fun_compile def main(): start_time = time.time() print(fun.calculate_pi(20_000_000)) end_time = time.time() print('Time:', end_time - start_time) start_time = time.time() print(fun_compile.calculate_pi(20_000_000)) end_time = time.time() print('Time:', end_time - start_time) start_time = time.time() print(fun_compile.calculate_pi_annotated(20_000_000)) end_time = time.time() print('Time:', end_time - start_time) if __name__ == '__main__': main()

运行后结果如下 可以看到加入变量类型加入后的代码运行效果更快了。