python - 用 C : Pass a list to PyArg_ParseTuple 扩展 python

Question

我一直在努力掌握使用 C 扩展 python，到目前为止，基于 documentation ，我在编写小型 C 函数并使用 Python 扩展它方面取得了相当大的成功。

但是，我现在遇到了一个相当简单的问题 - 我无法找到解决方案。所以，我想做的是将一个 double list 传递给我的 C 函数。例如，要传递 int，我会执行以下操作:

int squared(int n)
{
    if (n > 0)
        return n*n;
    else
        return 0;
}

static PyObject*
squaredfunc(PyObject* self, PyObject* args)
{
    int n;

    if (!PyArg_ParseTuple(args, "i", &n))
        return NULL;

    return Py_BuildValue("i", squared(n));
}

这会将 int n 毫无问题地传递给名为 squared 的 C 函数。

但是，如何将 list 传递给 C 函数？我确实尝试用谷歌搜索并阅读文档，但到目前为止，我还没有发现任何有用的东西。

如果有人能指出正确的方向，我将不胜感激。

谢谢。

Answer 1

PyArg_ParseTuple只能处理简单的 C 类型、复数、char *、PyStringObject *、PyUnicodeObject * 和 PyObject * .使用 PyListObject 的唯一方法是使用“O”的一些变体并将对象提取为 PyObject *。然后您可以使用 List Object API检查对象确实是一个列表 (PyList_Check)。然后您可以使用 PyList_Size 和 PyList_GetItem 迭代列表。请注意，在迭代时，您将获得 PyObject * 并且必须使用 floating point API访问实际值(通过执行 PyFloat_Check 和 PyFloat_AsDouble。)作为 List API 的替代方案，您可以更灵活地使用 iterator protocol (在这种情况下你应该只使用 PyIter_Check)。这将允许您迭代任何支持迭代器协议(protocol)的东西，例如列表、元组、集合等。

最后，如果你真的想让你的函数接受double n[]并且你想避免所有的手动转换，那么你应该使用像boost::python这样的东西。 .学习曲线和 API 更复杂，但 boost::python 会自动为您处理所有转换。

这是一个使用迭代器协议(protocol)进行循环的例子(这是未经测试的，你需要填写错误处理代码):

PyObject *obj;

if (!PyArg_ParseTuple(args, "O", &obj)) {
  // error
}

PyObject *iter = PyObject_GetIter(obj);
if (!iter) {
  // error not iterator
}

while (true) {
  PyObject *next = PyIter_Next(iter);
  if (!next) {
    // nothing left in the iterator
    break;
  }

  if (!PyFloat_Check(next)) {
    // error, we were expecting a floating point value
  }

  double foo = PyFloat_AsDouble(next);
  // do something with foo
}