通过放置一个列表中的每个第n个项目和另一个列表中的其他项目来合并Python中的列表？

Question

我有两个列表， list1和list2 。

这里len(list2) << len(list1) 。

现在我想合并两个列表，使得最终列表的每个第n个元素 来自list2 ， 而其他 元素 来自list1 。

例如：

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

list2 = ['x', 'y']

n = 3

现在最后的清单应该是：

['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']

实现这一目标的最恐怖的方法是什么？

我想将list2所有元素添加到最终列表中，最终列表应包括list1和list2所有元素。

Answer 1

使较大的列表成为迭代器可以轻松地为较小列表的每个元素获取多个元素：

list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
list2 = ['x', 'y'] 
n = 3

iter1 = iter(list1)
res = []
for x in list2:
    res.extend([next(iter1) for _ in range(n - 1)])
    res.append(x)
res.extend(iter1)

>>> res
['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']

这避免了insert ，这对于大型列表来说可能是昂贵的，因为每次需要重新创建整个列表。

Answer 2

要保留原始列表，可以尝试以下操作：

result = copy.deepcopy(list1)
index = n - 1
for elem in list2:
    result.insert(index, elem)
    index += n

结果

['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']

Answer 3

使用itertools模块和补充的more_itertools包 ，您可以通过几种不同的方式构建可迭代的解决方案。 首先进口：

import itertools as it, more_itertools as mt

第一个似乎是最干净的，但它依赖于more_itertools.chunked() 。

it.chain(*mt.roundrobin(mt.chunked(list1, n-1), list2))

这个只使用more_itertools.roundrobin() ，其实现来自itertools文档，因此如果您无权访问more_itertools您可以自己复制它。

mt.roundrobin(*([iter(list1)]*(n-1) + [list2]))

或者，这与第一个样本more_itertools而不使用任何more_itertools特定的函数。 基本上， grouper可以替代chunked ，但在某些情况下它会在最后添加None ，所以我将它包裹在其中。 it.takewhile去除它们。 当然，如果你在实际上包含None列表上使用它，它会在到达那些元素时停止，所以要小心。

it.takewhile(lambda o: o is not None,
   it.chain(*mt.roundrobin(mt.grouper(n-1, list1), list2))
)

我在Python 3.4上测试了这些，但我相信这些代码示例也适用于Python 2.7。

Answer 4

以下解决方案怎么样？ 但是我没有更好的...

>>> list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
>>> list2 = ['x', 'y']
>>> n = 2
>>> for i in range(len(list2)):
...     list1.insert(n, list2[i])
...     n += 3
...     
... 
>>> list1
['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']

n是2，因为列表中第三个元素的索引是2，因为它从0开始。

Answer 5

list(list1[i-1-min((i-1)//n, len(list2))] if i % n or (i-1)//n >= len(list2) else list2[(i-1)//n] for i in range(1, len(list1)+len(list2)+1))

绝对不是pythonic，但我认为在单行中进行它可能很有趣。 更可读（真的吗？）版本：

list(
    list1[i-1-min((i-1)//n, len(list2))]
        if i % n or (i-1)//n >= len(list2)
        else
    list2[(i-1)//n]
        for i in range(1, len(list1)+len(list2)+1)
)

基本上，一些修改索引并确定哪个列表和哪个索引从下一个元素。

Answer 6

另一种方法是，计算切片步骤：

list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
list2 = ['x', 'y'] 
n = 3

res = []
m = n - 1
start, end = 0, m
for x in list2:
    res.extend(list1[start:end])
    res.append(x)
    start, end = end, end + m
res.extend(list1[start:])

>>> res
['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']

Answer 7

list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
list2 = ['x', 'y']
n = 3
new = list1[:]

for index, item in enumerate(list2):
    new[n * (index + 1) - 1: n * (index + 1) - 1] = item

print(new)

Answer 8

我很佩服@David Z使用more_itertools 。 对工具的更新可以简化解决方案：

import more_itertools as mit

n = 3
groups = mit.windowed(list1, n-1, step=n-1)
list(mit.flatten(mit.interleave_longest(groups, list2)))
# ['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']

摘要： list2正在从list1交错到组中，最后扁平化为一个列表。

笔记

1. groups ： n-1大小的滑动窗口，例如[('a', 'b'), ('c', 'd'), ('e', 'f'), ('g', 'h')]
2. interleave_longest目前相当于roundrobin
3. None是默认填充值。 可选择使用filter(None, ...)删除filter(None, ...)

Answer 9

也许这是另一种解决方案，将list1切成正确的索引，然后将list2的元素添加到list1 。

>>> list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
>>> list2 = ['x', 'y'] 
>>> n = 3
>>> for i in range(len(list2)):
...     list1 = list1[:n*(i+1) - 1] + list(list2[i]) + list1[n*(i+1)-1:]
... 
>>> list1
['a', 'b', 'x', 'c', 'd', 'y', 'e', 'f', 'g', 'h']