Wie ein numpy Datensatz Array von C erstellen
Frage
Auf der Python-Seite kann ich neue numpy Rekord-Arrays erstellen wie folgt:
numpy.zeros((3,), dtype=[('a', 'i4'), ('b', 'U5')])
Wie mache ich das gleiche von einem C-Programm? Ich glaube, ich habe PyArray_SimpleNewFromDescr(nd, dims, descr)
zu nennen, aber wie konstruiere ich ein PyArray_Descr
, die für das Bestehen als drittes Argument ist angezeigt PyArray_SimpleNewFromDescr
?
Lösung
Mit PyArray_DescrConverter
. Hier ein Beispiel:
#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;
}
Danke href="https://stackoverflow.com/users/9530/adam-rosenfield"> Adam Rosenfield für den Hinweis auf Abschnitt 13.3.10 der Guide to NumPy .
Andere Tipps
Sehen Sie die zu NumPy Leitfaden, Abschnitt 13.3 .10. Es gibt viele verschiedene Möglichkeiten, einen Descriptor zu machen, obwohl es wie das Schreiben [('a', 'i4'), ('b', 'U5')]
nicht annähernd so einfach ist.