1

我有两个列表，我想以交替的方式组合它们，直到一个用完，然后我想继续从更长的列表中添加元素。

阿卡。

list1 = [a,b,c]

list2 = [v,w,x,y,z]

result = [a,v,b,w,c,x,y,z]

与这个问题类似（Pythonic way to combine two lists in an alternate way?），除了这些列表在第一个列表用完后停止组合:(。

4

5 回答 5

5

你可能对这个itertools食谱感兴趣：

def roundrobin(*iterables):
    "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
    # Recipe credited to George Sakkis
    pending = len(iterables)
    nexts = cycle(iter(it).next for it in iterables)
    while pending:
        try:
            for next in nexts:
                yield next()
        except StopIteration:
            pending -= 1
            nexts = cycle(islice(nexts, pending))

例如：

>>> from itertools import cycle, islice
>>> list1 = list("abc")
>>> list2 = list("uvwxyz")
>>> list(roundrobin(list1, list2))
['a', 'u', 'b', 'v', 'c', 'w', 'x', 'y', 'z']
于 2014-08-08T18:17:30.523 回答
5

这是来自优秀toolz的更简单的版本：

>>> interleave([[1,2,3,4,5,6,7,],[0,0,0]])
[1, 0, 2, 0, 3, 0, 4, 5, 6, 7]
于 2014-08-08T18:18:38.193 回答
1

您可以使用普通map和列表理解：

>>> [x for t in map(None, a, b) for x in t if x]
['a', 'v', 'b', 'w', 'c', 'x', 'y', 'z']
于 2014-08-08T18:24:58.877 回答
1

我的解决方案：

result = [i for sub in zip(list1, list2) for i in sub]

编辑：问题指定较长的列表应该在较短的列表的末尾继续，这个答案没有这样做。

于 2014-08-08T18:21:08.600 回答
0

我们可以将 zip_longest 与tcathcart答案一起使用

import itertools
result = [i for sub in itertools.zip_longest(list1, list2) for i in sub]
于 2022-02-24T13:02:10.360 回答