将文件句柄传递给cython函数

将文件句柄传递给cython函数,第1张

文件句柄传递给cython函数

定义类型

fid
将无济于事,因为调用python函数仍然很昂贵。尝试使用“
-a”标志编译示例以了解我的意思。但是,可以将低级C函数用于文件处理,以避免循环中的python开销。出于示例的考虑,我假设数据从文件的开头开始,并且其类型为
double

from libc.stdio cimport *cdef extern from "stdio.h":    FILE *fdopen(int, const char *)import numpy as npcimport numpy as npDTYPE = np.double # or whatever your type isctypedef np.double_t DTYPE_t # or whatever your type isdef FromFileSkip(fid, int count=1, int skip=0):    cdef int k    cdef FILE* cfile    cdef np.ndarray[DTYPE_t, ndim=1] data    cdef DTYPE_t* data_ptr    cfile = fdopen(fid.fileno(), 'rb') # attach the stream    data = np.zeros(count).astype(DTYPE)    data_ptr = <DTYPE_t*>data.data    # maybe skip some header bytes here    # ...    for k in range(count):        if fread(<void*>(data_ptr + k), sizeof(DTYPE_t), 1, cfile) < 0: break        if fseek(cfile, skip, SEEK_CUR): break    return data

请注意,的输出

cython -a example.pyx
显示循环内没有python开销。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5661877.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存