我们知道,在涉及到大量 CPU 计算的时候,Python 的运行效率可能不如其他语言。在之前的一篇文章:一日一技:立竿见影地把你的 Python 代码提速7倍中,我们讲到了如何通过把 Python 代码编译成 C 语言代码来提高代码的运行速度。今天,我们不用 C 语言,而用 Go 语言。并且,我们这次不是做转换,而是直接用 Python 调用 Go 语言写的代码。
今天的文章内容需要使用 Linux 和 macOS 系统实现。或者你也可以使用 Windows 10自带的 WSL2 Linux子系统。
今天我们的测试代码还是递归版本的斐波那契数列计算函数,由于递归版本涉及到大量重复计算,所以运行速度极慢。用 Python 计算第40项,耗时36.6秒,如下图所示:
如果直接使用 Golang 来写相同的算法,速度会远远快于 Python——同样是计算第40项,只需要0.5秒,如下图所示:
那么,我们能不能使用 Golang 写一个计算斐波那契数列的递归函数,但是使用 Python 来调用这个函数呢?
实际上是可以的,我们只需要把 Golang 写的程序编译为.so
文件就可以在 Python 里面调用了。
首先修改一下我们的 Golang 代码,把计算斐波拉契数列的函数fib
的首字母改成大写。在 Golang 中,只有首字母大写的函数,才能被package
外面的代码调用。
修改以后的代码如下:
需要注意的是,这里的注释//export Fib
不能省略,因为在编译成.so
文件的时候,编译器会寻找这个注释。同时,为了能正常编译,我们也必须导入C
这个包。
把代码保存为fib.go
,然后我们使用如下命令进行编译:
这条命令指定了编译模式为c-shared
,编译完成以后,当前文件夹下面会生成一个_fib.so
文件。这就是我们需要用 Python 载入的文件了。
下面,我们创建一个 Python 文件load.py
,代码如下:
在这段代码里面,通过 Python 自带的ctypes
模块,加载_fib.so
文件,然后提取出里面的Fib
函数。这个函数就是用递归方式计算斐波那契数列的函数了。同样是计算第40项,我们来看看耗时:
只需要0.5秒。耗时只有纯 Python 版本的 三十六分之一,也就是说,通过 Python 调用 Golang 的函数,速度是纯 Python 版本的72倍。
但需要注意的是,由于Golang 是静态强类型语言,定义函数的时候需要声明变量类型。如果你要导出的函数是字符串,那么变量类型需要改为*C.char
而不是简单的string
。这里涉及到 C 语言中的变量类型与 Golang 中变量类型的映射关系,可以参阅:cgo[1]。本文的例子中,由于整型比较特殊,直接使用int
就可以了。但其他的类型可能并没有这么简单。
end
Java遇见PythonGo
长按上方2秒,输入【面试】
【Java遇见PythonGo】 由一位资深老码农运营,混迹于世界500强多年的,做过开发,测试,项目管理,对技术比较痴迷,也是一位后端开发爱好者。
在这里希望给大家分享一些Java,Python和Go语言的编程技巧 ,以及趣味实战干货和面试经验,程序员圈的趣事。欢迎喜欢Java,Python和Go的程序员关注。
本篇文章来源于: 菜鸟学Python
本文为原创文章,版权归知行编程网所有,欢迎分享本文,转载请保留出处!
你可能也喜欢
- ♥ 如何在python中关闭线程11/01
- ♥ 为什么python可以多线程12/31
- ♥ python中if else如何判断表达式?12/26
- ♥ 如何理解python的递归函数?12/26
- ♥ Python 描述符中的三个方法是什么?11/08
- ♥ linux下如何卸载python09/05
内容反馈