![](/img/trans.png)
[英]Numpy C++: How to iterate over PyArrayObject without a segfault
[英]Numpy C extension: Access PyObjects in a PyArrayObject
我使用pd.DataFrame({“ A”:[“ dog”,“ cat”]})创建了一个熊猫数据框,并将其发送给ac扩展以生成numpy数组。 在C中,我这样做:
PyArrayObject * pArray = (PyArrayObject *) pArrayObj;
PyArray_Descr * pDesc = PyArray_DTYPE(pArray);
int kind = pDesc->kind;
npy_intp sz = PyArray_DIM(pArray, 0);
PyObject * p0 = (PyObject *) PyArray_GETPTR1(pArray, 0);
在上面的“种类”中包含“ O”(对于对象数组)和sz = 2(按预期)。 但是p0似乎没有访问第一个对象:
(gdb) p *p0
$10 = {ob_refcnt = 140737227161872, ob_type = 0x7ffff06e8180}
请注意,refcnt已损坏,ob_type无法解析(原本是PyUnicode_Type)。
有什么建议可以正确获取指向每个元素的指针?
我在Centos 7,G ++ 6.2.1上使用numpy 1.14
想通了怎么做:
npy_intp stride = PyArray_STRIDE(pArray, 0);
const char * dataptr = PyArray_BYTES(pArray);
PyObject * p1 = PyArray_GETITEM(pArray, dataptr);
PyObject * p2 = PyArray_GETITEM(pArray, dataptr + stride);
我发现了这个问题,因为我遇到了同样的问题。 原来的代码几乎是正确的,最后一行只需要更多的指针魔术:
PyArrayObject * pArray = (PyArrayObject *) pArrayObj;
PyObject * p0 = *(PyObject **) PyArray_GETPTR1(pArray, 0);
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.