提速72倍,在Python里面调用Golang函数

简介: 提速72倍,在Python里面调用Golang函数

摄影:产品经理上海味道不错的川菜

我们知道,在涉及到大量 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外面的代码调用。

修改以后的代码如下:

package main
import (    "C")
//export Fibfunc Fib(n int) int {    if n == 1 || n == 2{        return 1    } else {        return Fib(n - 1) + Fib(n - 2)    }}
func main() {}

需要注意的是,这里的注释//export Fib不能省略,因为在编译成.so文件的时候,编译器会寻找这个注释。同时,为了能正常编译,我们也必须导入C这个包。

把代码保存为fib.go,然后我们使用如下命令进行编译:


go build -buildmode=c-shared -o _fib.so fib.go

这条命令指定了编译模式为c-shared,编译完成以后,当前文件夹下面会生成一个_fib.so文件。这就是我们需要用 Python 载入的文件了。

下面,我们创建一个 Python 文件load.py,代码如下:


import ctypesimport time
so = ctypes.cdll.LoadLibrary('./_fib.so')fib = so.Fib
start = time.time()result = fib(40)end = time.time()print(f'斐波那契数列第40项:{result},耗时:{end - start}')

在这段代码里面,通过 Python 自带的ctypes模块,加载_fib.so文件,然后提取出里面的Fib函数。这个函数就是用递归方式计算斐波那契数列的函数了。同样是计算第40项,我们来看看耗时:

只需要0.5秒。耗时只有纯 Python 版本的 三十六分之一,也就是说,通过 Python 调用 Golang 的函数,速度是纯 Python 版本的72倍。

但需要注意的是,由于Golang 是静态强类型语言,定义函数的时候需要声明变量类型。如果你要导出的函数是字符串,那么变量类型需要改为*C.char而不是简单的string。这里涉及到 C 语言中的变量类型与 Golang 中变量类型的映射关系,可以参阅:cgo[1]。本文的例子中,由于整型比较特殊,直接使用int就可以了。但其他的类型可能并没有这么简单。

目录
相关文章
|
1天前
|
数据库 开发者 Python
实战指南:用Python协程与异步函数优化高性能Web应用
在快速发展的Web开发领域,高性能与高效响应是衡量应用质量的重要标准。随着Python在Web开发中的广泛应用,如何利用Python的协程(Coroutine)与异步函数(Async Functions)特性来优化Web应用的性能,成为了许多开发者关注的焦点。本文将从实战角度出发,通过具体案例展示如何运用这些技术来提升Web应用的响应速度和吞吐量。
8 1
|
1天前
|
调度 Python
揭秘Python并发编程核心:深入理解协程与异步函数的工作原理
在Python异步编程领域,协程与异步函数成为处理并发任务的关键工具。协程(微线程)比操作系统线程更轻量级,通过`async def`定义并在遇到`await`表达式时暂停执行。异步函数利用`await`实现任务间的切换。事件循环作为异步编程的核心,负责调度任务;`asyncio`库提供了事件循环的管理。Future对象则优雅地处理异步结果。掌握这些概念,可使代码更高效、简洁且易于维护。
6 1
|
24天前
|
Go
Golang语言之函数(func)进阶篇
这篇文章是关于Golang语言中函数高级用法的教程,涵盖了初始化函数、匿名函数、闭包函数、高阶函数、defer关键字以及系统函数的使用和案例。
17 3
Golang语言之函数(func)进阶篇
|
5天前
|
Python
[oeasy]python035_根据序号得到字符_chr函数_字符_character_
本文介绍了Python中的`ord()`和`chr()`函数。`ord()`函数通过字符找到对应的序号,而`chr()`函数则根据序号找到对应的字符。两者互为逆运算,可以相互转换。文章还探讨了单双引号在字符串中的作用,并解释了中文字符和emoji也有对应的序号。最后总结了`ord()`和`chr()`函数的特点,并提供了学习资源链接。
14 4
|
24天前
|
Go
Golang语言之函数(func)基础篇
这篇文章深入讲解了Golang语言中函数的定义和使用,包括函数的引入原因、使用细节、定义语法,并通过多个案例展示了如何定义不返回任何参数、返回一个或多个参数、返回值命名、可变参数的函数,同时探讨了函数默认值传递、指针传递、函数作为变量和参数、自定义数据类型以及返回值为切片类型的函数。
20 2
Golang语言之函数(func)基础篇
|
8天前
|
Java Python
全网最适合入门的面向对象编程教程:50 Python函数方法与接口-接口和抽象基类
【9月更文挑战第18天】在 Python 中,虽无明确的 `interface` 关键字,但可通过约定实现类似功能。接口主要规定了需实现的方法,不提供具体实现。抽象基类(ABC)则通过 `@abstractmethod` 装饰器定义抽象方法,子类必须实现这些方法。使用抽象基类可使继承结构更清晰、规范,并确保子类遵循指定的方法实现。然而,其使用应根据实际需求决定,避免过度设计导致代码复杂。
|
11天前
|
Python
全网最适合入门的面向对象编程教程:Python函数方法与接口-函数与方法的区别和lamda匿名函数
【9月更文挑战第15天】在 Python 中,函数与方法有所区别:函数是独立的代码块,可通过函数名直接调用,不依赖特定类或对象;方法则是与类或对象关联的函数,通常在类内部定义并通过对象调用。Lambda 函数是一种简洁的匿名函数定义方式,常用于简单的操作或作为其他函数的参数。根据需求,可选择使用函数、方法或 lambda 函数来实现代码逻辑。
|
24天前
|
Python
python 函数
【9月更文挑战第4天】python 函数
37 5
|
1天前
|
开发者 Python
Python 中的 Input 函数及其实现机制
Python 中的 Input 函数及其实现机制
|
24天前
|
Unix Go
Golang语言标准库time之日期和时间相关函数
这篇文章是关于Go语言日期和时间处理的文章,介绍了如何使用Go标准库中的time包来处理日期和时间。
29 3