python定义一个数组_python – 使用C API创建一个numpy自定义类对象数组

由于Quaternion不是直接数字类型,因此您的数组必须将numpy.object作为dtype.因此,您可以使用PyArray_SimpleNew(…,NPY_OBJECT)来创建数组并填充数据.

问题是你的Quaternion类不是python类型.因此,使用对Quaternion类型的对象的引用来填充数组将不起作用. (在这种情况下,如果从python中填充四元数的数组中提取元素,您会发生什么？)

相反,您需要使用PyQuaternion之类的东西包装Quaternion类.包装器负责引用计数和内存管理.它看起来像：

typedef struct {

PyObject_HEAD

Quaternion *q;

}PyQuaternion;

static PyTypeObject PyQuaternion_Type = {

PyObject_HEAD_INIT(NULL)

0, /*ob_size*/

"Quaternion", /*tp_name*/

sizeof(PyQuaternion), /*tp_basicsize*/

/* ... */

};

static PyObject *

PyQuaternion_new(PyTypeObject *type, PyObject *args, PyObject *kwds){

/* ... */

};

static int

PyQuaternion_init(PyQuaternion *self, PyObject *args, PyObject *kwds){

/* ... */

};

static void PyQuaternion_dealloc(PyQuaternion *self){

/* ... */

};

此外,您可以为PyQuaternionType定义自己的C-API,允许您从Quaternions创建PyQuaternions

static PyObject *

PyQuaternion_New(Quaternion *q){

PyQuaternion *self;

self = (PyQuaternion *)PyQuaternion_Type.tp_new(type, NULL, NULL);

self->q = q;

return (PyObject *)self;

}

请注意,自我> q将由PyQuaternion_dealloc函数处理,因此请考虑内存管理.最简单的方法是将所有权传递给包装器,让PyQuaternion_dealloc解除分配自我> q.

PyQuaternion_New函数允许您包装Quaternion对象并将它们填充到任何python容器中,如列表,元组,当然还有ntype = numpy.object的numpy数组.