NumPy C-API：将类型对象转换为类型编号

Question

功能

PyObject* PyArray_TypeObjectFromType(int);

将NumPy标量类型（NPY_BOOL，NPY_BYTE，...）的类型编号转换为相应的类型对​​象。

你如何进行相反的转换，从NumPy标量类型的类型对象到相应的类型号？

编辑：以下代码基于kwatford的答案。 它接受类型对象（如int和numpy.int16）和字符串（如“int”，u“int”和“int16”）。

int numpyScalarTypeNumber(PyObject* obj)
{
    PyArray_Descr* dtype;
    if(!PyArray_DescrConverter(obj, &dtype)) return NPY_NOTYPE;
    int typeNum = dtype->type_num;
    Py_DECREF(dtype);
    return typeNum;
}

Answer 1

如果你可以得到PyArray_Descr结构而不是PyArray_TypeObject ，你可以简单地看一下type_num字段。 可以使用PyArray_DescrFromType通过类型编号获取描述结构。 如果你看一下这个链接，还有一些函数可以将各种东西转换成描述。 它们通常比类型对象更有用，它们也包含对其类型的引用。