Golang調用Python


https://yq.aliyun.com/articles/117329

Python是時髦的機器學習御用開發語言,Golang是大紅大紫的新時代后端開發語言。Python很適合讓搞算法的寫寫模型,而Golang很適合提供API服務,兩位同志都紅的發紫,這里就介紹一下正確攪基的辦法

原理

Python提供了豐富的C-API。而C和Go又可以通過cgo無縫集成。所以,直接通過Golang調用libpython,就可以實現Go調Python的功能了。確實沒啥神奇,只要會用C調Python,馬上就知道怎么用了。但問題是,如果有的選擇,這個年代還有多少人願意去裸寫C和C++呢?誠心默念Golang大法好。

准備工作

  • Python :確保Python正確安裝,所謂正確安裝,就是在系統中能找到libpython.so(dylib),找到Python.h。一般linux直接安裝python-devel,mac直接用homebrew安裝就可以。
  • Golang安裝:Golang不需要什么特殊的處理,能找到go即可。

安裝libpython-go-binding

雖然直接用cgo調用libpython也不是不可以,但是有native-binding用起來肯定要爽的多。Github上有一個現成的Binding庫go-python

 go get github.com/sbinet/go-python

如果Python安裝正確,這里會自動編譯並顯示提示,事就這樣成了。

Have a try

首先寫一個測試Python腳本

import numpy import sklearn a = 10 def b(xixi): return xixi + "haha"

然后寫一個Go腳本:

package main import ( "github.com/sbinet/go-python" "fmt" ) func init() { err := python.Initialize() if err != nil { panic(err.Error()) } } var PyStr = python.PyString_FromString var GoStr = python.PyString_AS_STRING func main() { // import hello InsertBeforeSysPath("/Users/vonng/anaconda2/lib/python2.7/site-packages") hello := ImportModule("/Users/vonng/Dev/go/src/gitlab.alibaba-inc.com/cplus", "hello") fmt.Printf("[MODULE] repr(hello) = %s\n", GoStr(hello.Repr())) // print(hello.a) a := hello.GetAttrString("a") fmt.Printf("[VARS] a = %#v\n", python.PyInt_AsLong(a)) // print(hello.b) b := hello.GetAttrString("b") fmt.Printf("[FUNC] b = %#v\n", b) // args = tuple("xixi",) bArgs := python.PyTuple_New(1) python.PyTuple_SetItem(bArgs, 0, PyStr("xixi")) // b(*args) res := b.Call(bArgs, python.Py_None) fmt.Printf("[CALL] b('xixi') = %s\n", GoStr(res)) // sklearn sklearn := hello.GetAttrString("sklearn") skVersion := sklearn.GetAttrString("__version__") fmt.Printf("[IMPORT] sklearn = %s\n", GoStr(sklearn.Repr())) fmt.Printf("[IMPORT] sklearn version = %s\n", GoStr(skVersion.Repr())) } // InsertBeforeSysPath will add given dir to python import path func InsertBeforeSysPath(p string) string { sysModule := python.PyImport_ImportModule("sys") path := sysModule.GetAttrString("path") python.PyList_Insert(path, 0, PyStr(p)) return GoStr(path.Repr()) } // ImportModule will import python module from given directory func ImportModule(dir, name string) *python.PyObject { sysModule := python.PyImport_ImportModule("sys") // import sys path := sysModule.GetAttrString("path") // path = sys.path python.PyList_Insert(path, 0, PyStr(dir)) // path.insert(0, dir) return python.PyImport_ImportModule(name) // return __import__(name) }

打印輸出為:

repr(hello) = <module 'hello' from '/Users/vonng/Dev/go/src/gitlab.alibaba-inc.com/cplus/hello.pyc'> a = 10 b = &python.PyObject{ptr:(*python._Ctype_struct__object)(0xe90b1b8)} b('xixi') = xixihaha sklearn = <module 'sklearn' from '/Users/vonng/anaconda2/lib/python2.7/site-packages/sklearn/__init__.pyc'> sklearn version = '0.18.1'

這里簡單解釋一下。首先將這個腳本的路徑添加到sys.path中。然后調用PyImport_ImportModule導入包

使用GetAttrString可以根據屬性名獲取對象的屬性,相當於python中的.操作。調用Python函數可以采用Object.Call方法,,列表參數使用Tuple來構建。返回值用PyString_AS_STRING從Python字符串轉換為C或Go的字符串。

更多用法可以參考Python-C API文檔

但是只要有這幾個API,就足夠 Make python module rock & roll。充分利用Golang和Python各自的特性,構建靈活而強大的應用了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM