题
在Python方面,我可以创建新的numpy记录数组,如下所示:
numpy.zeros((3,), dtype=[('a', 'i4'), ('b', 'U5')])
如何从C程序中执行相同的操作?我想我必须调用 PyArray_SimpleNewFromDescr(nd,dims,descr)
,但是如何构造一个 PyArray_Descr
,它适合作为第三个参数传递给 PyArray_SimpleNewFromDescr 代码>
解决方案
使用 PyArray_DescrConverter
。这是一个例子:
#include <Python.h>
#include <stdio.h>
#include <numpy/arrayobject.h>
int main(int argc, char *argv[])
{
int dims[] = { 2, 3 };
PyObject *op, *array;
PyArray_Descr *descr;
Py_Initialize();
import_array();
op = Py_BuildValue("[(s, s), (s, s)]", "a", "i4", "b", "U5");
PyArray_DescrConverter(op, &descr);
Py_DECREF(op);
array = PyArray_SimpleNewFromDescr(2, dims, descr);
PyObject_Print(array, stdout, 0);
printf("\n");
Py_DECREF(array);
return 0;
}
感谢 Adam Rosenfield 指向 NumPy指南。
其他提示
请参阅 NumPy指南,第13.3节0.10。有很多不同的方法可以创建一个描述符,虽然它不像编写 [('a','i4'),('b','U5')]
那么容易。
不隶属于 StackOverflow