Como criar uma matriz de registro numpy de C
Pergunta
No lado do Python, posso criar novas matrizes de registro numpy da seguinte forma:
numpy.zeros((3,), dtype=[('a', 'i4'), ('b', 'U5')])
Como posso fazer o mesmo a partir de um programa C? Acho que tem que chamar PyArray_SimpleNewFromDescr(nd, dims, descr)
, mas como faço para construir um PyArray_Descr
que é apropriado para passar como o terceiro argumento para PyArray_SimpleNewFromDescr
?
Solução
Use PyArray_DescrConverter
. Aqui está um exemplo:
#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;
}
Graças à Adam Rosenfield para apontar a Seção 13.3.10 do Guide to NumPy .
Outras dicas
Veja a Guide to NumPy , seção 13.3 .10. Há muitas maneiras diferentes de fazer um descritor, embora não seja tão fácil como escrever [('a', 'i4'), ('b', 'U5')]
.
Licenciado em: CC-BY-SA com atribuição
Não afiliado a StackOverflow