两个列表之间的公共元素不使用Python中的集合

时间:2010-04-28 07:53:25

标签: python list set

我想要计算两个列表的相同元素。列表可以有重复的元素,所以我无法将其转换为集合并使用&操作

a=[2,2,1,1]
b=[1,1,3,3]

设置(a)&设置(b)工作
a& b不起作用

有可能使用setoud和dictonary吗?

3 个答案:

答案 0 :(得分:11)

在Python 3.x(以及Python 2.7,发布时),您可以使用collections.Counter

>>> from collections import Counter
>>> list((Counter([2,2,1,1]) & Counter([1,3,3,1])).elements())
[1, 1]

这是使用collections.defaultdict的替代方法(Python 2.5及更高版本中提供)。它有一个很好的属性,结果的顺序是确定的(它基本上对应于第二个列表的顺序)。

from collections import defaultdict

def list_intersection(list1, list2):
    bag = defaultdict(int)
    for elt in list1:
        bag[elt] += 1

    result = []
    for elt in list2:
        if elt in bag:
            # remove elt from bag, making sure
            # that bag counts are kept positive
            if bag[elt] == 1:
                del bag[elt]
            else:
                bag[elt] -= 1
            result.append(elt)

    return result

对于这两种解决方案,输出列表中任何给定元素x的出现次数是两个输入列表中x出现次数的最小值。从你的问题不清楚这是否是你想要的行为。

答案 1 :(得分:8)

使用集合是最有效的,但您始终可以r = [i for i in l1 if i in l2]

答案 2 :(得分:0)

SilentGhost,Mark Dickinson和Lo'oris是对的,非常感谢您报告此问题 - 我需要列表的常见部分,因此:

A = [1,1,1,2]

B = [1,1,3,3]

结果应为[1,1]

很抱歉在不合适的地方发表评论 - 我今天已经注册了。

我修改了你的解决方案:

def count_common(l1,l2):
        l2_copy=list(l2)
        counter=0
        for i in l1:
            if i in l2_copy:
                counter+=1
                l2_copy.remove(i)
        return counter

l1=[1,1,1]
l2=[1,2]
print count_common(l1,l2)

1