用Pythonic方式混合两个列表

Question

我有两个长度为n和n + 1的列表：

[a_1, a_2, ..., a_n]
[b_1, b_2, ..., b_(n+1)]

我想要一个函数，给出一个包含两个元素的列表，即

[b_1, a_1, ..., b_n, a_n, b_(n+1)]

以下工作原理，但看起来并不聪明：

def list_mixing(list_long,list_short):
    list_res = []
    for i in range(len(list_short)):
        list_res.extend([list_long[i], list_short[i]])
    list_res.append(list_long[-1])
    return list_res

谁能建议一种更Python化的方式来做到这一点？ 谢谢！

Answer 1

>>> import itertools
>>> a
['1', '2', '3', '4', '5', '6']
>>> b
['a', 'b', 'c', 'd', 'e', 'f']
>>> list(itertools.chain.from_iterable(zip(a,b)))
['1', 'a', '2', 'b', '3', 'c', '4', 'd', '5', 'e', '6', 'f']

zip()产生一个具有最短参数长度的可迭代对象。 您可以将a[-1]附加到结果中，也可以将itertools.zip_longest （对于Python 2.x使用izip_longest）与填充值一起使用，然后删除该值。

您可以在此解决方案中使用两个以上的输入序列。

对于不附加最后一个值的情况，您可以尝试使用这种肮脏的方法，但是我并不推荐这样做，目前尚不清楚：

>>> a
[1, 2, 3, 4, 5]
>>> b
['a', 'b', 'c', 'd', 'e', 'f']
>>> [a[i//2] if i%2 else b[i//2] for i in range(len(a)*2+1)]
['a', 1, 'b', 2, 'c', 3, 'd', 4, 'e', 5, 'f']

（对于Python 2.x，请使用single / ）

Answer 2

恕我直言，最好的方法是：

result = [item for sublist in zip(a,b) for item in sublist]

它比求和和归约的方法还快。

UPD对不起，您的第二个列表大了一个元素：)还有另一种疯狂的方式：

result = [item for sublist in map(None, a, b) for item in sublist][:-1]

Answer 3

>>> long = [1, 3, 5, 7]
>>> short = [2, 4, 6]
>>> mixed = []
>>> for i in range(len(long)):
>>>     mixed.append(long[i])
>>>     if i < len(short)
>>>         mixed.append(short[i])
>>> mixed
[1, 2, 3, 4, 5, 6, 7]

Answer 4

混合两个列表是zip的工作：

res = []
for a,b in zip(list_long, list_short):
    res += [a,b]

对于不同长度的列表，请定义您自己的函数：

def mix(list_long, list_short):
    result = []
    i,j = iter(list_long), iter(list_short)
    for a,b in zip(i,j):
        res += [a,b]
    for rest in i:
        result += rest
    for rest in j:
        result += rest
    return result

使用Mihail给出的答案 ，我们可以将其简化为：

def mix(list_long, list_short):
    i,j = iter(list_long), iter(list_short)
    result = [item for sublist in zip(i,j) for item in sublist]
    result += [item for item in i]
    result += [item for item in j]
    return result

Answer 5

我会结合使用以上答案：

>>> a = ['1', '2', '3', '4', '5', '6']

>>> b = ['a', 'b', 'c', 'd', 'e', 'f', 'g']

>>> [i for l in izip_longest(a, b, fillvalue=object) for i in l if i is not object]
<<< ['1', 'a', '2', 'b', '3', 'c', '4', 'd', '5', 'e', '6', 'f', 'g']

Answer 6

使用izip_longest填补列表中没有的内容，或者不想保留结果。 如果您不想保留任何解析为False ，请使用izip_longest和filter的默认值：

from itertools import chain, izip_longest
l1 = [1,3,5]
l2 = [2,4,6,8,10]
filter(None, chain(*izip_longest(l1,l2)))

结果是： [1, 2, 3, 4, 5, 6, 8, 10]

使用None填补空白并使用filter将其移除：

filter(lambda x: x is not None, chain(*izip_longest(l1,l2, fillvalue=None)))

为了提高效率，当l1或l2不是短列表而是很长或无限的可迭代对象时，请使用ifilter代替filter ，它会为您提供可迭代对象，而不是将所有内存都放在列表中。 例：

from itertools import chain, izip_longest, ifilter
for value in ifilter(None, chain(*izip_longest(iter1,iter1))):
    print value

Answer 7

sum([[x,y] for x,y in zip(b,a)],[])+[b[-1]]

注意：这仅适用于给定的列表长度，但可以轻松扩展到任意长度的列表。

Answer 8

这是我发现的最好的：

import itertools

l1 = [1, 3, 5]
l2 = [2, 4, 6, 8, 10]

result = [
    x # do something
    for x in itertools.chain.from_iterable(itertools.zip_longest(l1, l2, l1))
    if x is not None
]
result
# [1, 2, 1, 3, 4, 3, 5, 6, 5, 8, 10]

为了更加清楚， zip_longest按索引将元素分组在一起：

iter = list(itertools.zip_longest(l1, l2, l1))
iter[0]
# (1, 2, 1)
iter[1]
# (3, 4, 3)
iter[-1] # last
# (None, 10, None)

之后， itertools.chain.from_iterable它们按顺序展平。

最好的原因：

• 滤波None ，不推荐了，使用列表理解
• 列表理解还可以让您立即使用x来“做某事”
• 当某些列表比最短的列表长时，它不会丢弃项目
• 它适用于任何数量的列表
• 实际上，很容易推断出发生了什么事，而与那里所有的“聪明”相反

Answer 9

more-itertools具有roundrobin ，可以准确地完成此工作：

from more_itertools import roundrobin

l1 = [1, 3, 5]
l2 = [2, 4, 6, 8, 10]

print(list(roundrobin(l1,l2)))
# [1, 2, 1, 3, 4, 3, 5, 6, 5, 8, 10]

Answer 10

您可以执行以下操作（假设len(list_long)==len(list_short)+1 ：

def list_mixing(list_long,list_short):
    return [(list_long[i/2] if i%2==0 else list_short[i/2]) for i in range(len(list_long)+len(list_short)]

我在哪里使用/进行整数除法（确切的运算符取决于语言版本）。

Answer 11

使用邮编。 这将为您提供一个元组列表，例如：[（'a_1'，'b_1'），（'a_2'，'b_2'），（'a_3'，'b_3'）]

如果要将其清理成一个不错的列表，只需使用enumerate遍历元组列表即可：

alist = ['a_1', 'a_2', 'a_3']
blist = ['b_1', 'b_2', 'b_3']
clist = []

for i, (a, b) in enumerate(zip(alist, blist)):
    clist.append(a)
    clist.append(b)
print clist
['a_1', 'b_1', 'a_2', 'b_2', 'a_3', 'b_3']