Question

一开始，我有2个列表和1个列表，说明我应该以什么顺序合并这两个列表。例如，我的第一个列表等于[a, b, c]，第二个列表等于[d, e]和＆＃39;合并＆＃39;列表等于[0, 1, 0, 0, 1]。

这意味着：首先制作合并列表我需要从第一个列表中获取元素，然后是第二个，然后是第一个，然后是第一个，然后是第二个......我最终得到[a, d, b, c, e]。为了解决这个问题，我只使用了循环和两个＆＃34;指针＆＃34;，但我想知道我是否可以做更多pythonic这个任务...我试图找到一些可以帮助我的功能，但没有真正的结果。

Answer 1

您可以从这些列表创建迭代器，遍历排序列表，并在其中一个迭代器上调用next：

i1 = iter(['a', 'b', 'c'])
i2 = iter(['d', 'e'])
# Select the iterator to advance: `i2` if `x` == 1, `i1` otherwise
print([next(i2 if x else i1) for x in [0, 1, 0, 0, 1]]) # ['a', 'd', 'b', 'c', 'e']

可以将此解决方案推广到任意数量的列表，如下所示

def ordered_merge(lists, selector):
    its = [iter(l) for l in lists]
    for i in selector:
        yield next(its[i])

In [4]: list(ordered_merge([[3, 4], [1, 5], [2, 6]], [1, 2, 0, 0, 1, 2]))
Out[4]: [1, 2, 3, 4, 5, 6]

如果排序列表包含字符串，浮点数或任何其他不能用作列表索引的对象，请使用字典：

def ordered_merge(mapping, selector):
    its = {k: iter(v) for k, v in mapping.items()}
    for i in selector:
        yield next(its[i])

In [6]: mapping = {'A': [3, 4], 'B': [1, 5], 'C': [2, 6]}

In [7]: list(ordered_merge(mapping, ['B', 'C', 'A', 'A', 'B', 'C']))
Out[7]: [1, 2, 3, 4, 5, 6]

当然，你也可以使用整数作为字典键。

或者，您可以逐个从每个原始列表的左侧删除元素，并将它们添加到结果列表中。快速举例：

In [8]: A = ['a', 'b', 'c']
   ...: B = ['d', 'e']
   ...: selector = [0, 1, 0, 0, 1]
   ...: 

In [9]: [B.pop(0) if x else A.pop(0) for x in selector]
Out[9]: ['a', 'd', 'b', 'c', 'e']

我希望第一种方法更有效率（list.pop(0) 慢）。

Answer 2

这个怎么样，

list1 = ['a', 'b', 'c']
list2 = ['d', 'e']
options = [0,1,0,0,1] 

list1_iterator = iter(list1)
list2_iterator = iter(list2)

new_list = [next(list2_iterator) if option else next(list1_iterator) for option in options]

print(new_list)
# Output
['a', 'd', 'b', 'c', 'e']

使用给定的合并顺序合并两个或多个列表

2 个答案: