之前一篇文章里提到了利用Cython來(lái)編譯Python，這次來(lái)講一下如何用Cython給Python寫擴(kuò)展庫(kù)。
兩種語(yǔ)言混合編程，其中最重要的是類型的傳遞。
我們用一個(gè)簡(jiǎn)單的例子進(jìn)行入門：這次的目標(biāo)是用C語(yǔ)言寫一個(gè)Numpy的加法和元素相乘模塊。在本例中，Numpy的array被傳入到C語(yǔ)言模塊內(nèi)，變成了二維數(shù)組。
1. 頭文件main.h：
#ifndef _MAIN_H
#define _MAIN_H
void plus(double *a, double *b, double *r, int n, int m); // 矩陣加法
void mul(double *a, double *b, double *r, int n, int m); // 矩陣按元素相乘
void main(double *a, double *b, double *r, int n, int m, int times); // 用于測(cè)試的main函數(shù)
#endif
2.? 把主要代碼寫在main.c中：
#include "main.h"
/***********************************
* 矩陣的加法
* 利用數(shù)組是順序存儲(chǔ)的特性, *
* 通過(guò)降維來(lái)訪問(wèn)二維數(shù)組! *
* r
***********************************/
void plus(double *a, double *b, double *r, int n, int m)
{
int i, j;
for(i = 0; i < n; i++)
{
for(j = 0; j < m; j++)
*(r + i*m + j) = *(a + i*m + j) + *(b + i*m + j);
}
}
/***********************************
* 矩陣的按元素乘法
* 利用數(shù)組是順序存儲(chǔ)的特性, *
* 通過(guò)降維來(lái)訪問(wèn)二維數(shù)組! *
* r
***********************************/
void mul(double *a, double *b, double *r, int n, int m)
{
int i, j;
for(i = 0; i < n; i++)
{
for(j = 0; j < m; j++)
*(r + i*m + j) = *(a + i*m + j) * *(b + i*m + j);
}
}
/***********************************
* main函數(shù)
* 利用數(shù)組是順序存儲(chǔ)的特性, *
* 通過(guò)降維來(lái)訪問(wèn)二維數(shù)組! *
* r
***********************************/
void main(double *a, double *b, double *r, int n, int m, int times)
{
int i;
// 循環(huán)times次
#pragma omp parallel for
for (i = 0; i < times; i++)
{
// 矩陣的加法
plus(a, b, r, n, m);
// 矩陣按元素相乘
mul(a, b, r, n, m);
}
}
這個(gè)main.c中實(shí)現(xiàn)了矩陣的加法、矩陣按元素相乘的功能，用到的數(shù)據(jù)結(jié)構(gòu)是二維數(shù)組，但是因?yàn)镃語(yǔ)言中給函數(shù)傳遞二維數(shù)組比較麻煩，這里用降維的方法實(shí)現(xiàn)。另外在main()函數(shù)中，采用一個(gè)循環(huán)來(lái)進(jìn)行測(cè)試，以測(cè)試性能。
3. 下面編寫test.pyx文件來(lái)調(diào)用上述C函數(shù)（注意，后綴是.pyx噢）：詳細(xì)的知識(shí)點(diǎn)在注釋中寫出來(lái)了~
# 既要import numpy, 也要用cimport numpy
import time
import numpy as np
cimport numpy as np
# 使用Numpy-C-API
np.import_array()
# cdefine C 函數(shù)
cdef extern from "main.h":
void plus(double *a, double *b, double *r, int n, int m)
void mul(double *a, double *b, double *r, int n, int m)
void main(double *a, double *b, double *r, int n, int m, int times)
"""
# 定義一個(gè)"包裝函數(shù)", 用于調(diào)用C語(yǔ)言的main函數(shù)，調(diào)用范例：plus_fun(a, b, r)
# 在這里要注意函數(shù)傳入的參數(shù)的類型聲明，double表示數(shù)組的元素是double類型的，
# ndim = 2表示數(shù)組的維度是2
# 在調(diào)用main函數(shù)時(shí)，要把python的變量強(qiáng)制轉(zhuǎn)化成相應(yīng)的類型（以確保無(wú)誤），比如
# 當(dāng)然，基本類型如int，可以不顯式地寫出來(lái)，如下面的a.shape[0]、a.shape[1]
"""
def main_func(np.ndarray[double, ndim=2, mode="c"] a not None,
np.ndarray[double, ndim=2, mode="c"] b not None,
np.ndarray[double, ndim=2, mode="c"] r not None,
times not None):
main(
np.PyArray_DATA(a),
np.PyArray_DATA(b),
np.PyArray_DATA(r),
a.shape[0],
a.shape[1],
times)
4. 為了用Cython編譯上述代碼，我們創(chuàng)建一個(gè)setup.py文件：
import numpy
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
filename = 'test' # 源文件名
full_filename = 'test.pyx' # 包含后綴的源文件名
setup(
name = 'test',
cmdclass = {'build_ext': build_ext},
ext_modules=[Extension(filename,sources=[full_filename, "main.c"],
include_dirs=[numpy.get_include()])],
)
5. 上述的main.h、main.c、test.pyx一定要放在同一個(gè)文件夾下。此時(shí)在該文件夾下按住shift鍵，然后右擊鼠標(biāo)，打開(kāi)cmd或PowerShell控制臺(tái)，在控制臺(tái)中運(yùn)行以下命令進(jìn)行Cython編譯：
python setup.py build_ext --i
或者：
python setup.py build_ext --inplace
編譯成功的圖例：
此時(shí)在同目錄下會(huì)生成“test.cp36-win_amd64.pyd”的二進(jìn)制碼文件，它是閉源的，但是可以直接用python來(lái)import。下面編寫測(cè)試代碼main.py來(lái)進(jìn)行測(cè)試：
import test
import time
import numpy as np
start_time = time.time()
a = np.random.rand(100, 100) * 2 - 1 # 生成300*300的隨即矩陣
b = np.random.rand(100, 100) * 2 - 1
r = np.empty_like(a) # 創(chuàng)建一個(gè)空矩陣，用來(lái)存儲(chǔ)計(jì)算結(jié)果
test.main_func(a, b, r, 500000) # 調(diào)用main_func進(jìn)行測(cè)試
end_time = time.time()
print(end_time - start_time) # 輸出時(shí)間
print(r) # 輸出運(yùn)行結(jié)果
執(zhí)行結(jié)果：
通過(guò)本例我們可以看到：將循環(huán)放在C語(yǔ)言模塊中，而不是原生的Python中，可以提高執(zhí)行效率。
以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
更多文章、技術(shù)交流、商務(wù)合作、聯(lián)系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號(hào)聯(lián)系： 360901061
您的支持是博主寫作最大的動(dòng)力，如果您喜歡我的文章，感覺(jué)我的文章對(duì)您有幫助，請(qǐng)用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點(diǎn)擊下面給點(diǎn)支持吧，站長(zhǎng)非常感激您！手機(jī)微信長(zhǎng)按不能支付解決辦法：請(qǐng)將微信支付二維碼保存到相冊(cè)，切換到微信，然后點(diǎn)擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對(duì)您有幫助就好】元
