Question

我正在尝试编写一个Python函数，它将两个列表作为参数并对它们进行交错。应保留组件列表的顺序。如果列表的长度不同，则较长列表的元素最终应该在结果列表的结尾。例如，我想把它放在Shell中：

interleave(["a", "b"], [1, 2, 3, 4])

然后回复：

["a", 1, "b", 2, 3, 4]

如果你能帮助我，我会很感激。

Answer 1

这是我如何使用itertools模块的各个位来实现的。它适用于任意数量的迭代，而不仅仅是两个：

from itertools import chain, izip_longest # or zip_longest in Python 3
def interleave(*iterables):

    sentinel = object()
    z = izip_longest(*iterables, fillvalue = sentinel)
    c = chain.from_iterable(z)
    f = filter(lambda x: x is not sentinel, c)

    return list(f)

Answer 2

你可以试试这个：

In [30]: from itertools import izip_longest

In [31]: l = ['a', 'b']

In [32]: l2 = [1, 2, 3, 4]

In [33]: [item for slist in izip_longest(l, l2) for item in slist if item is not None]
Out[33]: ['a', 1, 'b', 2, 3, 4]

izip_longest'将两个列表一起'拉链'，但不是停留在最短列表的长度，而是一直持续到最长的列表耗尽：

In [36]: list(izip_longest(l, l2))
Out[36]: [('a', 1), ('b', 2), (None, 3), (None, 4)]

然后，您可以通过迭代压缩列表中每对中的每个项目来添加项目，省略值为None的项目。正如@Blckknight所指出的，如果您的原始列表已经有None值，这将无法正常工作。如果在您的情况下可行，则可以使用fillvalue的{{1}}属性来填充izip_longest以外的内容（正如@Blckknight在他的回答中所做的那样）。

以上示例为函数：

None

Answer 3

一个不是很优雅的解决方案，但仍然可能会有所帮助

def interleave(lista, listb):
    (tempa, tempb) = ([i for i in reversed(lista)], [i for i in reversed(listb)])
    result = []
    while tempa or tempb:
        if tempa:
            result.append(tempa.pop())
        if tempb:
            result.append(tempb.pop())

    return result

或单行

   def interleave2(lista, listb):
    return reduce(lambda x,y : x + y,
                  map(lambda x: x[0] + x[1],
                      [(lista[i:i+1], listb[i:i+1])
                       for i in xrange(max(len(lista),len(listb)))]))

Answer 4

另一种解决方案基于：我将如何手工完成？好吧，几乎是手工使用内置的zip()，并扩展了压缩的结果较长列表的长度由较长的列表的尾部组成：

#!python2

def interleave(lst1, lst2):
    minlen = min(len(lst1), len(lst2))        # find the length of the shorter
    tail = lst1[minlen:] + lst2[minlen:]      # get the tail
    result = []
    for t in zip(lst1, lst2):                 # use a standard zip
        result.extend(t)                      # expand tuple to two items
    return result + tail                      # result of zip() plus the tail

print interleave(["a", "b"], [1, 2, 3, 4])
print interleave([1, 2, 3, 4], ["a", "b"])
print interleave(["a", None, "b"], [1, 2, 3, None, 4])

打印结果：

['a', 1, 'b', 2, 3, 4]
[1, 'a', 2, 'b', 3, 4]
['a', 1, None, 2, 'b', 3, None, 4]

在python v.2中交换两个不同长度的列表？

4 个答案: