cython程序例子（fib）

使用Cython加速Python：从编写到编译

最新推荐文章于 2024-05-29 11:06:26 发布

原创最新推荐文章于 2024-05-29 11:06:26 发布 · 483 阅读

0 ·

CC 4.0 BY-SA版权

Cython 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了如何使用Cython将Python代码编译为C扩展模块，以提高程序性能。首先展示了Cython代码示例，定义了一个计算斐波那契数列的函数。接着，通过`setup.py`配置文件，利用`cythonize`将Cython代码转换为C代码，并编译为`.so`动态链接库文件。最后，演示了如何在Python环境中导入并使用这个加速后的模块，输出斐波那契数列的第20项以及函数的文档字符串。

1）编写cython程序

# fib.pyx
def fib(n):
    """这是一个扩展模块"""
    cdef int i
    cdef double a = 0.0, b = 1.0
    for i in range(n):
        a, b = a + b, a
	return a

2）编译so文件

# setup.py
from distutils.core import setup
from Cython.Build import cythonize

setup(ext_modules=cythonize("fib.pyx"))

# cythonize("fib.pyx")负责将Cython代码转成C代码
# 然后根据C代码生成扩展模块，我们可以传入单个文件，也可以是多个文件组成的列表
# 或者一个glob模式，会匹配满足模式的所有Cython文件

编译命令：

python setup.py build_ext --inplace

–inplace 表示在在当前目录下生成.so文件。

编译之后目录中增加文件：

fib.cpython-36m-x86_64-linux-gnu.so 和 fib.c。
在这里插入图片描述

3）调用一下

import fib
# 我们看到该pyd文件直接就被导入了，至于中间的是对应的解释器版本、操作系统等信息
# 可以不用管，甚至你删掉只保留fib.so也是可以的。
print(fib) # 打印文件路径
print(fib.fib(20))  # 6765.0
print(fib.fib.__doc__)  # 打印注释