python - 嵌套列表上的 Itertools 链

Question

我有两个列表按顺序组合以创建一个具有 python map 和 zip 功能的嵌套列表；但是，我希望用 itertools 重新创建它。

此外，我试图理解为什么当我插入两个列表时 itertools.chain 返回一个扁平列表，但是当我添加一个嵌套列表时它只返回嵌套列表。

任何有关这两个问题的帮助将不胜感激。

from itertools import chain

a = [0,1,2,3]
b = [4,5,6,7]

#how can I produce this with itertools?
c = list(map(list, zip(a,b)))
print(c) #[[0, 4], [1, 5], [2, 6], [3, 7]]

d = list(chain(c))
print(d) #[[0, 4], [1, 5], [2, 6], [3, 7]]

d = list(chain(a,b))
print(d) #[0, 1, 2, 3, 4, 5, 6, 7]

Answer 1

我会尽力回答您的问题。

首先，itertools.chain并不像你想象的那样工作。 chain需要x可迭代的数量并按顺序迭代它们。当您调用chain时，它本质上(内部)将对象打包到一个列表中:

chain("ABC", "DEF") # Internally creates ["ABC", "DEF"]

在方法内部，它一次访问其中的每一项，并迭代它们:

for iter_item in arguments:
    for item in iter_item:
        yield item

所以当你调用chain([[a,b],[c,d,e],[f,g]])时，它创建一个包含一个可迭代对象的列表:您作为参数传递的列表。所以现在看起来像这样:

[ #outer
    [ #inner
        [a,b],
        [c,d,e],
        [f,g]
    ]
]

chain因此迭代 inner 列表，并返回三个元素: [a,b] , [c,d,e] ，和[f,g]为了。然后它们被 list 重新包装。 ，给你你最初拥有的东西。

顺便说一句，有一种方法可以做你想做的事:chain.from_iterable 。这是 chain 的备用构造函数它接受单个可迭代对象，例如您的列表，并将元素拉出来进行迭代。所以代替这个:

# chain(l)
[ #outer
    [ #inner
        [a,b],
        [c,d,e],
        [f,g]
    ]
]

你得到这个:

# chain.from_iterable(l)
[
    [a,b],
    [c,d,e],
    [f,g]
]

这将迭代三个子列表，并按一个顺序返回它们，因此 list(chain.from_iterable(l))将返回[a,b,c,d,e,f,g] .

关于你的第二个问题:虽然我不知道为什么itertools是此过程所必需的，您可以在 Python 2.x 中执行此操作:

list(itertools.izip(x,y))

但是，在 3.x 中，izip功能已被删除。还有zip_longest ，它将匹配尽可能多的对，并接受额外对的填充值: list(zip_longest([a,b,c],[d,e,f,g,h],fillvalue="N"))返回[(a,d),(b,e),(c,f),(N,g),(N,h)]因为第二个列表比第一个列表长。正常 zip将采用最短的迭代并切断其余的。

换句话说，除非你想要 zip_longest而不是zip , itertools没有内置的压缩​​方法。