如何在Numpy C API中使用PyArray_SearchSorted

Question

在C擴展中，我正在訪問傳遞給函數的兩個數組：

PyObject *xw_array = PyArray_FROM_OTF(xw_obj, NPY_DOUBLE, NPY_IN_ARRAY);
PyObject *x1_array = PyArray_FROM_OTF(x1_obj, NPY_DOUBLE, NPY_IN_ARRAY);

然后我想對這兩個數組使用PyArray_SearchSorted我目前正在做：

PyObject *ix_array = PyArray_SearchSorted(xw_array, x1_array);

但這會導致以下錯誤：

propagate_pure.c:123:138: error: too few arguments to function call, expected 4, have 2
PyObject *ix_array = (*(PyObject * (*)(PyArrayObject *, PyObject *, NPY_SEARCHSIDE, PyObject *)) PyArray_API[131])(xw_array, x1_array);
                     ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~                   ^

使用PyArray_SearchSorted的正確方法是什么？ 需要四個參數？ 該文檔僅提及兩個。

Answer 1

PyArray_SearchSorted的完整聲明在這里 ：

NPY_NO_EXPORT PyObject *
PyArray_SearchSorted(PyArrayObject *op1, PyObject *op2,
                     NPY_SEARCHSIDE side, PyObject *perm)

您需要提供一個side和一個perm參數。 默認NPY_SEARCHLEFT和NULL 。 因此，以下應該工作：

PyObject *ix_array = PyArray_SearchSorted(xw_array, x1_array,
                                          NPY_SEARCHLEFT, NULL);