时间戳

首页 > 解决方案 > 列表理解中的python可迭代列表理解

python - 列表理解中的python可迭代列表理解

问题描述

我试图弄清楚为什么我的嵌套列表理解不起作用。代码：

def myzip(*args):
    try:
        x = [iter(i) for i in [*args]]
        while True:
        # for i in range(5):
            yield [next(ii) for ii in x]
            # I want to know why the commented line below won't work
            # yield [next(ii) for ii in [iter(i) for i in [*args]]]
            # or why this outputs [1, 'a']
            # return [next(ii) for ii in [iter(i) for i in [*args]]]
    except:
        pass


def main():
    print(list(myzip([1, 2, 3, 4, 5], ['a', 'b', 'c'])))

if __name__ == '__main__':
    main()

如果我分配[iter(i) for i in [*args]]给x然后使用这个列表理解[next(ii) for ii in x]一切正常。输出将是：

[[1, 'a'], [2, 'b'], [3, 'c']]

但是，如果我尝试省略变量x并在一个更大的列表理解（注释行）中执行[next(ii) for ii in [iter(i) for i in [*args]]]它，它会进入无限循环。如果将无限循环替换为 for 循环（注释行），则输出为：

[[1, 'a'], [1, 'a'], [1, 'a'], [1, 'a'], [1, 'a']]

此外，如果我尝试return [next(ii) for ii in [iter(i) for i in [*args]]]它只会返回：

[1, 'a']

有人能告诉我这是为什么吗？

标签： pythonlist-comprehension

解决方案

所以我不会直接回答你的问题（因为评论很好地涵盖了这一点），但我将展示如何将其作为一个可怕的单线来完成，只是为了好玩：

def my_zip(*args):
    yield from iter(lambda data=[iter(a) for a in args]: [next(d) for d in data], None)

有三个部分。

  • iter, after的第一次使用yield from创建了一个callable_iterator. 这将重复调用它的第一个参数（the lambda），直到它看到它的第二个参数（sentinel 值，在这种情况下None）或者它StopIteration遇到任何异常时遇到错误。（在这种情况下，这将是一个StopIteration例外。
  • 在lambdai 中有一个默认值，data它是您要迭代的迭代器的列表。this 被创建一次，因此每次对该函数的调用都将引用 this。
  • 最后，在正文中lambda，我在每个函数调用中手动推进每个迭代器data一次。这将一直持续到其中一个迭代器用尽，此时StopIteration将引发 a。正如您所看到的，在这种情况下，哨兵值无关紧要，因为第一次iter调用将在它得到它时立即保释，StopIteration这将冒泡到list消费者，导致它停止迭代。由于异常，哨兵值无关紧要。

最后：请永远不要在实际代码中实际执行此操作。

编辑：一些示例输出

In [90]: list(my_zip('abcd', [1,2,3]))
Out[90]: [['a', 1], ['b', 2], ['c', 3]]

In [91]: list(my_zip('abcd', [1,2,3,4,5,6]))
Out[91]: [['a', 1], ['b', 2], ['c', 3], ['d', 4]]

解释编辑：向@superb rain 致敬，感谢他们在下面的聪明评论。

推荐阅读