Jak utworzyć tablicę z rekordową numpy C

Question

po stronie Python, mogę tworzyć nowe tablice rekordowe numpy następująco:

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

Jak mogę zrobić to samo z programu C? Przypuszczam, że muszę zadzwonić pod numer PyArray_SimpleNewFromDescr(nd, dims, descr), ale jak skonstruować PyArray_Descr, który jest odpowiedni do przekazania jako trzeci argument do PyArray_SimpleNewFromDescr?

Answer 1

Użyj PyArray_DescrConverter. Oto przykład:

#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; 
} 

Dzięki Adam Rosenfield dla wskazując na punkcie 13.3.10 z Guide to NumPy.

Answer 2

Zobacz Guide to NumPy, sekcja 13.3.10. Istnieje wiele różnych sposobów tworzenia deskryptorów, chociaż nie jest to tak łatwe jak pisanie [('a', 'i4'), ('b', 'U5')].