迁移Python C扩展至NumPy 2：替换PyArray_GetCastFunc处理字节交换数组

针对你在迁移Python C扩展到NumPy 2时遇到的字节交换数组处理问题，以下是适配NumPy 2 API的逐元素处理方案，核心思路是保留原有的逐元素流处理逻辑，用C强制转换替代已移除的PyArray_GetCastFunc，同时调整copyswap函数的获取方式：

核心步骤说明 #

1. 字节交换处理：NumPy 2中copyswap函数需通过PyDataType_GetArrFuncs(descr)->copyswap获取，功能与NumPy 1.x一致——可在复制元素时完成字节交换，得到机器序的原始类型值。
2. 类型转换替代：对于常规数值类型（int/float系列），直接通过C强制转换将机器序的原始值转为double用于运算；运算结果再转换为输出数组的dtype，最后通过copyswap写入字节交换格式的输出数组。

完整代码示例 #

#include <numpy/arrayobject.h>

// 示例：处理单个(i,j)位置的元素，输入输出均为2D字节交换数组
int process_element(PyObject* input_frame_1, PyObject* output_frame, int i, int j) {
    // ---------------------- 输入处理部分 ----------------------
    PyArrayObject* arr_in = (PyArrayObject*)input_frame_1;
    PyArray_Descr* descr_in = PyArray_DESCR(arr_in);
    PyArrFuncs* funcs_in = PyDataType_GetArrFuncs(descr_in, NPY_ARRAY_NOTSWAPPED);
    PyArray_CopySwapFunc* swap_in = funcs_in->copyswap;
    bool need_swap_in = PyArray_ISBYTESWAPPED(arr_in);

    // 临时存储机器序的输入元素（足够容纳常见数值类型）
    char temp_in[8];
    double val_double = 0.0;

    // 获取输入元素的指针
    char* src_ptr = (char*)PyArray_GETPTR2(arr_in, i, j);
    // 字节交换并复制到临时变量（得到机器序值）
    swap_in(src_ptr, temp_in, 1, need_swap_in);

    // 转换为double类型
    switch (descr_in->type_num) {
        case NPY_INT8: val_double = *(int8_t*)temp_in; break;
        case NPY_UINT8: val_double = *(uint8_t*)temp_in; break;
        case NPY_INT16: val_double = *(int16_t*)temp_in; break;
        case NPY_UINT16: val_double = *(uint16_t*)temp_in; break;
        case NPY_INT32: val_double = *(int32_t*)temp_in; break;
        case NPY_UINT32: val_double = *(uint32_t*)temp_in; break;
        case NPY_INT64: val_double = *(int64_t*)temp_in; break;
        case NPY_UINT64: val_double = *(uint64_t*)temp_in; break;
        case NPY_FLOAT32: val_double = *(float*)temp_in; break;
        case NPY_FLOAT64: val_double = *(double*)temp_in; break;
        default:
            PyErr_SetString(PyExc_TypeError, "Unsupported input dtype");
            return -1;
    }

    // ---------------------- 运算逻辑 ----------------------
    // 示例：这里替换为你的均值、中位数等运算
    double result = val_double; // 仅作示例，实际为运算结果

    // ---------------------- 输出处理部分 ----------------------
    PyArrayObject* arr_out = (PyArrayObject*)output_frame;
    PyArray_Descr* descr_out = PyArray_DESCR(arr_out);
    PyArrFuncs* funcs_out = PyDataType_GetArrFuncs(descr_out, NPY_ARRAY_NOTSWAPPED);
    PyArray_CopySwapFunc* swap_out = funcs_out->copyswap;
    bool need_swap_out = PyArray_ISBYTESWAPPED(arr_out);

    char temp_out[8];
    // 将运算结果转换为输出dtype的机器序值
    switch (descr_out->type_num) {
        case NPY_INT8: *(int8_t*)temp_out = (int8_t)result; break;
        case NPY_UINT8: *(uint8_t*)temp_out = (uint8_t)result; break;
        case NPY_INT16: *(int16_t*)temp_out = (int16_t)result; break;
        case NPY_UINT16: *(uint16_t*)temp_out = (uint16_t)result; break;
        case NPY_INT32: *(int32_t*)temp_out = (int32_t)result; break;
        case NPY_UINT32: *(uint32_t*)temp_out = (uint32_t)result; break;
        case NPY_INT64: *(int64_t*)temp_out = (int64_t)result; break;
        case NPY_UINT64: *(uint64_t*)temp_out = (uint64_t)result; break;
        case NPY_FLOAT32: *(float*)temp_out = (float)result; break;
        case NPY_FLOAT64: *(double*)temp_out = result; break;
        default:
            PyErr_SetString(PyExc_TypeError, "Unsupported output dtype");
            return -1;
    }

    // 写入输出数组，完成字节交换（如果需要）
    char* dst_ptr = (char*)PyArray_GETPTR2(arr_out, i, j);
    swap_out(temp_out, dst_ptr, 1, need_swap_out);

    return 0;
}

关键注意事项 #

• copyswap函数的参数：swap(src, dst, n, swap_flag)，当swap_flag为true时，复制时自动完成字节交换，确保dst中是机器序的原始类型值；false则直接复制。
• 强制转换仅适用于常规数值类型，如果涉及自定义dtype，需要额外实现类型转换逻辑。
• 逐元素处理的方式避免了全数组载入内存转换字节序的问题，适配你的大数量数组场景。

内容的提问来源于stack exchange，提问作者Sergio