我想要计算两个列表的相同元素。列表可以有重复的元素,所以我无法将其转换为集合并使用&操作
a=[2,2,1,1]
b=[1,1,3,3]
设置(a)&设置(b)工作
a& b不起作用
有可能使用setoud和dictonary吗?
答案 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