Как создать массив пустых записей из C
Вопрос
На стороне Python я могу создавать новые массивы записей следующим образом:
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;
}
Спасибо Адаму Розенфилду за указание на раздел 13.3.10 Руководство по NumPy .
Другие советы
См. Руководство по NumPy , раздел 13.3. +0,10. Есть много разных способов сделать дескриптор, хотя это не так просто, как написать [('a', 'i4'), ('b', 'U5')]
.
Не связан с StackOverflow