Comment créer un tableau numpy record à partir de C
Question
Du côté de Python, je peux créer de nouveaux tableaux d’enregistrements numpy comme suit:
numpy.zeros((3,), dtype=[('a', 'i4'), ('b', 'U5')])
Comment puis-je faire la même chose avec un programme C? Je suppose que je dois appeler PyArray_SimpleNewFromDescr (nd, dims, descr)
, mais comment puis-je construire un PyArray_Descr
approprié pour passer comme troisième argument à PyArray_SimpleNewFromDescr
?
La solution
Utilisez PyArray_DescrConverter
. Voici un exemple:
#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;
}
Merci à Adam Rosenfield d'avoir indiqué la section 13.3.10 de la Guide de NumPy .
Autres conseils
Voir le Guide de NumPy , section 13.3. .dix. Il existe de nombreuses façons de créer un descripteur, bien que ce ne soit pas aussi simple que d'écrire [('a', 'i4'), ('b', 'U5')]
.
Licencié sous: CC-BY-SA avec attribution
Non affilié à StackOverflow