• 15.20 处理C语言中的可迭代对象
    • 问题
    • 解决方案
    • 讨论

    15.20 处理C语言中的可迭代对象

    问题

    你想写C扩展代码处理来自任何可迭代对象如列表、元组、文件或生成器中的元素。

    解决方案

    下面是一个C扩展函数例子，演示了怎样处理可迭代对象中的元素：

    1. static PyObject *py_consume_iterable(PyObject *self, PyObject *args) {
    2.   PyObject *obj;
    3.   PyObject *iter;
    4.   PyObject *item;
    5.  
    6.   if (!PyArg_ParseTuple(args, "O", &obj)) {
    7.     return NULL;
    8.   }
    9.   if ((iter = PyObject_GetIter(obj)) == NULL) {
    10.     return NULL;
    11.   }
    12.   while ((item = PyIter_Next(iter)) != NULL) {
    13.     /* Use item */
    14.     ...
    15.     Py_DECREF(item);
    16.   }
    17.  
    18.   Py_DECREF(iter);
    19.   return Py_BuildValue("");
    20. }

    讨论

    本节中的代码和Python中对应代码类似。PyObject_GetIter() 的调用和调用 iter() 一样可获得一个迭代器。PyIter_Next() 函数调用 next 方法返回下一个元素或NULL(如果没有元素了)。要注意正确的内存管理—— Py_DECREF() 需要同时在产生的元素和迭代器对象本身上同时被调用，以避免出现内存泄露。

    原文:

    http://python3-cookbook.readthedocs.io/zh_CN/latest/c15/p20_consuming_an_iterable_from_c.html