NumPy C-API: chuyển đổi loại đối tượng để gõ số

Question

Chức năng

PyObject* PyArray_TypeObjectFromType(int); 

chuyển số kiểu cho một vô hướng loại NumPy (NPY_BOOL, NPY_BYTE, ...) cho các đối tượng loại tương ứng.

Làm cách nào để bạn thực hiện chuyển đổi ngược lại, từ đối tượng loại cho loại vô hướng NumPy sang số loại tương ứng?

Chỉnh sửa: Mã sau dựa trên câu trả lời của kwatford. Nó chấp nhận cả hai đối tượng kiểu như int và numpy.int16 và các chuỗi như "int", u "int" và "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

Nếu bạn có thể nhận PyArray_Descr cấu trúc hơn là PyArray_TypeObject s, bạn chỉ có thể nhìn vào các lĩnh vực type_num. Các cấu trúc descr có thể được mua thông qua số kiểu sử dụng PyArray_DescrFromType. Nếu bạn nhìn vào liên kết đó, cũng có một vài hàm nữa để chuyển đổi nhiều thứ khác nhau thành các cấu trúc descr. Chúng có thể hữu ích hơn nói chung so với các đối tượng kiểu, và chúng chứa các tham chiếu đến các kiểu của chúng.