C中处理Python的Numpy数组的代码范例

int fext_python_action(const ImageBuffer* pImage, DataBuffer* pOutput)
{
  PyObject* pArgs = PyTuple_New(１);
  PyTuple_SetItem(pArgs, 0, Py_BuildValue("i", 1));
  PyObject* pResult = PyObject_CallObject(g_pPythonFunction_FextAction, pArgs);
    //开始处理返回值。具体含义看注释。通过运行测试。
    PyArrayObject* pContArray = PyArray_GETCONTIGUOUS((PyArrayObject*)pResult);
    float* pDataPtr = (float*)PyArray_DATA(pContArray);
    //数组多少个维度？
    int num_dim = PyArray_NDIM(pContArray);
    //每个维度的长度
    npy_intp* pDim = PyArray_DIMS(pContArray);
    //得到元素总数
    int nDataCount = 1;
    for (int i = 0; i < num_dim; i++)
    {
        nDataCount *= pDim[i];
    }
    GH_LOG_INFO("num_dim=%d, nDataCount=%d", num_dim, nDataCount);
    //输出前4个
    for (int i = 0; i < 4; i++)
    {
        GH_LOG_INFO("data[%d]=%f", i, pDataPtr[i]);
    }
    //输出后4个
    for (int i = nDataCount-4; i < nDataCount; i++)
    {
        GH_LOG_INFO("data[%d]=%f", i, pDataPtr[i]);
    }
  Py_DECREF(pContArray);
  Py_DECREF(pResult);
  Py_DECREF(pArgs);
  return 0;
}