合并两个无与伦比的分类元素列表,将其相对顺序保存在Python3中



我有两个有序列表,其中有无与伦比的元素:

l1 = [1, 2, 3, 4]  
l2 = [obj1, obj2, obj3]  

我想将它们随机合并到一个列表中,但要保留它们的相对顺序,例如:

ml = [1, 2, obj1, obj2, 3, obj3, 4]  

ml = [obj1, 1, 2, 3, obj2, obj3, 4]  

如果有人帮助我,我会很感激。

简单函数可修改输入并返回列表:

import random
l1 = [1, 2, 3, 4]
l2 = ["obj1", "obj2", "obj3"]
def merge_random(lists):
    res = []
    while lists:
        li = random.choice(lists)
        res.append(li.pop(0))
        lists = [li for li in lists if li]
    return res
print(merge_random((l1, l2)))

示例输出:

['obj1', 'obj2', 1, 2, 'obj3', 3, 4]

发电机,越多的Pythonic解决方案

Jon Clements建议了这种解决方案。它利用发电机,不会修改用户输入。显然,这是解决问题的更多的Pythonic解决方案。

import random
def f(*iterables):
    todo = [iter(it) for it in iterables]
    while todo:
        idx = random.randrange(len(todo))
        try:
            yield next(todo[idx])
        except StopIteration:
            del todo[idx]

可以这样使用:

l1 = [1, 2, 3, 4]
l2 = ["obj1", "obj2", "obj3"]
for i in f(l1, l2):
    print(i)

示例输出:

obj1
1
2
obj2
obj3
3
4

最新更新