当使用ctypes数组作为numpy数组时,PEP 3118会发出警告

时间:2011-02-10 23:51:44

标签: python numpy warnings ctypes pep3118

当我尝试将ctypes数组用作numpy数组时,我收到以下警告消息:

Python 2.7.1 (r271:86832, Nov 27 2010, 18:30:46) [MSC v.1500 32 bit (Intel)] on
win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import ctypes, numpy
>>> TenByteBuffer = ctypes.c_ubyte * 10
>>> a = TenByteBuffer()
>>> b = numpy.ctypeslib.as_array(a)
C:\Python27\lib\site-packages\numpy\ctypeslib.py:402: RuntimeWarning: Item size
computed from the PEP 3118 buffer format string does not match the actual item s
ize.
  return array(obj, copy=False)
>>> b
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0], dtype=uint8)

但是代码似乎正在起作用。忽略这个警告是不是一个坏主意?

背景: 我正在调用一个实时生成数据的C DLL。我需要传递DLL一系列缓冲区来保存数据。在等待下一个缓冲区填充时,我想用numpy处理最新的缓冲区并保存结果。我正在使用上面的代码生成缓冲区,似乎工作正常,但我不想在地毯下扫描一个重要的问题。

2 个答案:

答案 0 :(得分:11)

这是Python中的一个错误。 ctypes目前生成无效的PEP 3118类型代码,Numpy注意到: http://bugs.python.org/issue10746 http://bugs.python.org/issue10744

当存在这种不一致时,Numpy会跳过使用PEP 3118缓冲区接口,并回退到旧的(过时的)缓冲区接口。这应该可以正常工作。

您可以使用Python的warnings模块使警告静音。但是,警告可能会对性能产生影响。

您还可以通过将ctypes对象包装在buffer()中来尝试解决此问题。

答案 1 :(得分:3)

有一种更方便的方法,可以完全避免警告:

不是首先将数据创建为ctypes数组,然后将其转换为NumPy数组,而是立即将其创建为NumPy数组,然后在ctypes原型中使用numpy.ctypeslib.ndpointer作为类型说明符。举个例子,假设您有一个名为f的C函数,它以char*size_t为参数:

void f(char* buf, size_t len);

你的ctypes原型将是

from numpy.ctypeslib import ndpointer
some_dll = ctypes.CDLL(...)
some_dll.f.argtypes = [ndpointer(numpy.uint8, flags="C_CONTIGUOUS"),
                       ctypes.c_size_t]
some_dll.f.restype = None

您可以将此功能称为

a = numpy.zeros(10, numpy.uint8)
f(a, a.size)