Python的itertools乘积内存消耗

Question

documentation表示笛卡尔积函数

the actual implementation does not build up intermediate results in memory.

生成器怎么可能呢？有人可以给我举个例子吗2 个生成器的内存消耗有限？

Answer 1

查看模块的源代码，itertools.product() 实际上将每个参数转换为一个元组:

// product_new() in itertoolsmodule.c
for (i=0; i < nargs ; ++i) {
    PyObject *item = PyTuple_GET_ITEM(args, i);
    PyObject *pool = PySequence_Tuple(item); //<==== Call tuple(arg)
    if (pool == NULL)
        goto error;
    PyTuple_SET_ITEM(pools, i, pool);
    indices[i] = 0;
}

换句话说，itertools.product() 的内存消耗似乎与输入参数的大小成线性关系。