Как создать массив пустых записей из C - PullRequest
8 голосов
/ 18 октября 2008

На стороне Python я могу создать новые массивы записей следующим образом:

numpy.zeros((3,), dtype=[('a', 'i4'), ('b', 'U5')])

Как мне сделать то же самое из программы на C? Я полагаю, мне нужно вызвать PyArray_SimpleNewFromDescr(nd, dims, descr), но как мне создать PyArray_Descr, который подходит для передачи в качестве третьего аргумента PyArray_SimpleNewFromDescr?

Ответы [ 2 ]

10 голосов
/ 18 октября 2008

Используйте 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;
}

Благодарю Адама Розенфилда за указание на раздел 13.3.10 Руководства по NumPy .

5 голосов
/ 18 октября 2008

См. Руководство по NumPy , раздел 13.3.10. Есть много разных способов сделать дескриптор, хотя это не так просто, как написать [('a', 'i4'), ('b', 'U5')].

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...