如何在两个列表中形成匹配元素列表,包括重复?

时间:2015-02-19 14:11:49

标签: python list set duplicates match

我有两个清单:

string1Elements = ['down', 'down', 'down', 'down']
string2Elements = ['down', 'down', 'right', 'down']

我想形成两个列表共有的元素列表,包括重复列表。我想要的结果如下:

['down', 'down', 'down']

考虑我想到的匹配的方法如下:

0:状态如下:

string1Elements = ['down', 'down', 'down', 'down']
string2Elements = ['down', 'down', 'right', 'down']

两个列表的大小相同,因此可以任意选择一个。如果列表的大小不同,则会选择较短的列表。

1:获取string1Elements的第一个元素(所选列表)。这个元素是否存在于string2Elements中?如果是,请将其附加到匹配列表中,然后从string1Elementsstring2Elements 删除

string1Elements = ['down', 'down', 'down']
string2Elements = ['down', 'right', 'down']
matches         = ['down']

2:获取string1Elements的第一个元素。这个元素是否存在于string2Elements中?如果是,请将其附加到匹配列表中,然后将其从string1Elementsstring2Elements中删除。

string1Elements = ['down', 'down']
string2Elements = ['right', 'down']
matches         = ['down', 'down']

3:获取string1Elements的第一个元素。这个元素是否存在于string2Elements中?如果是,请将其附加到匹配列表中,然后将其从string1Elementsstring2Elements中删除。

string1Elements = ['down']
string2Elements = ['right']
matches         = ['down', 'down', 'down']

4:获取string1Elements的第一个元素。这个元素是否存在于string2Elements中?如果是,请将其附加到匹配列表中,然后将其从string1Elementsstring2Elements中删除。

string1Elements = ['down']
string2Elements = ['right']
matches         = ['down', 'down', 'down']

5:已检查所有元素。


上述过程只是为了解释我想如何处理重复元素。我实际上不想更改列表string1Elementsstring2Elements

由于重复的元素,我认为不能以明显的方式使用集合:

matches = list(set(string2Elements).intersection(string1Elements))

我尝试使用列表理解进行快速测试:

matches = [element for element in string1Elements if element in string2Elements]

这些方法都不够。我怎么能按照我描述的方式实现匹配?

1 个答案:

答案 0 :(得分:1)

如果匹配,您可以从一个列表系统地弹出并附加到结果列表。因为您已从列表中弹出,所以它不会将list1中的一个元素与list2中的五个元素匹配,因为它会被弹出。一个示例函数:

def intersect(a, b):
    if len(b) < len(a):  # iff b is shorter than a
        a, b = b, a      # swap the lists.
    b = b[:]  # To prevent modifying the lists
    return [b.pop(b.index(i)) for i in a if i in b]

用法:

list1 = ['down', 'down', 'down', 'down']
list2 = ['down', 'down', 'right', 'down']
matches = intersect(list1, list2)
print(" ".join(matches))

# Prints:
down down down

这完全与OP解释如何在问题中完成,除了它只删除较短列表中的元素。
在多个列表上工作的一个可以实现如下

def multi_intersect(*args):
    if len(args) == 1:
        try:
            return multi_intersect(*args)
        except TypeError:
            pass
    inters = [item for sublist in args for item in sublist]
    for arg in args:
        inters = intersect(inters, arg)
    return inters