Cython 入门

Act enthusiastic and you will be enthusiastic.

1.Cython 的由来

Python 是一门开源免费、语法简单的语言,然而解释性语言的通病就是“运行速度慢”,它远慢于 C/C++。为了解决该问题,Python 官网提供了「Python/C API」,可以实现用 C 语言编写 Python 库,即通过 C 语言来编写部分核心逻辑来提高整体性能。然而,「Python/C API」门槛太高,于是诞生了 Cython。
只需编写简单的 Cython(语法与 Python 基本一致)代码,就可以实现 C 扩展,如下图所示:

2.什么是 Cython ?

「Cython」 是一个针对 Python 语言和扩展的 Cython 语言(基于 Pyrex)的优化的静态编译器,即可将 Python/Cython 编译成 C。它使得为 Python 编写 C 扩展就像 Python 本身一样简单,优点包括:

  • 1)既具备了 Python 快速开发的特点,
  • 2)又可以让代码运行起来像 C 一样快,
  • 3)同时还可以方便地调用 C library。

在计算科学中,常用 Cython 来加速 Python 代码。

3.安装 Cython

3.1)配置 gcc

Cython 需要一个 C 编译器,不同平台的安装配置:

  • windows
    首先,安装 MingW-w64 编译器,命令:conda install libpython m2w64-toolchain -c msys2
    然后,在 Python 安装路径下的 /Lib/distutils 下创建 distutils.cfg,并写入如下内容:
    [build] compiler=mingw32
  • macOS
    安装 XCode 即可。
  • linux:
    gcc 一般都是配置好的,如果没有可执行命令:sudo apt-get install build-essential
3.2)安装 Cython 库

命令:pip install cython
Anaconda 环境下安装命令:conda install cython

查看版本:cython --version

4.简单示例

1)定义要加速的模块。

原始 Python 文件 multiply_python.py

from time import time

# 自定义"计时"装饰器
def cal_time(func):
    def cal_time_wrapper(*args):
        start_time = time()
        result = func(*args)
        end_time = time()
        print("Python 耗时(秒):", end_time - start_time)
        return result
    return cal_time_wrapper

# 添加"计时"装饰器
@cal_time
def my_multiply(a, b):
    return a * b

需要加速的 Cython/Python 模块 multiply_cython.pyx

from time import time

# 自定义"计时"装饰器
def cal_time(func):
    def cal_time_wrapper(*args):
        start_time = time()
        result = func(*args)
        end_time = time()
        print("Cython 耗时(秒):", end_time - start_time)
        return result
    return cal_time_wrapper

# 添加"计时"装饰器
@cal_time
def my_multiply(a, b):
    return a * b
2)编写 setup.py(名字不限) 文件——指定扩展模块,内容如下:
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext

ext_modules = [Extension("multiply_cython", ["multiply_cython.pyx"])]
setup(
    name='multiply_demo',
    cmdclass={'build_ext': build_ext},
    ext_modules=ext_modules
)
3)执行命令生成 C 扩展:python setup.py build_ext --inplace

执行成功,生成一个 .c 和一个 .so 文件:

4)编写 test.py 进行测试,内容如下:
import numpy as np
import multiply_python
import multiply_cython

x = np.arange(1, 25).reshape(2, 3, 4)
y = np.arange(10, 34).reshape(2, 3, 4)
multiply_python.my_multiply(x, y)
multiply_cython.my_multiply(x, y)

本例中,Cython 加速后的速度是原来的4.75倍,运行结果如下:

Python 耗时(秒): 9.059906005859375e-06
('Cython 耗时(秒):', 1.9073486328125e-06)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,039评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,223评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,916评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,009评论 1 291
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,030评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,011评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,934评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,754评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,202评论 1 309
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,433评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,590评论 1 346
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,321评论 5 342
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,917评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,568评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,738评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,583评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,482评论 2 352

推荐阅读更多精彩内容