Python求含重复元素的两个列表的差集问题

嘿，这个问题我见过好多次啦——普通的集合差集或者简单列表推导式都会忽略元素的重复次数，而你要的是带重复元素的多重集合差集。下面给你两种靠谱的解决方案，尤其是第一个能完美保留原列表顺序的方法，应该正好符合你的需求：

问题根源 #

普通的集合操作（比如set(list1) - set(list2)）会自动去重，而简单的列表推导式（比如[x for x in list1 if x not in list2]）会把所有在list2里出现过的元素都过滤掉，不管list1里出现了几次。这就是为什么你只能得到['E', 'O']，而漏掉了list1里的第二个'N'——因为list2里有一个'N'，普通方法会把list1里的两个'N'都排除。

方法1：保持原列表顺序的遍历法 #

这个方法会逐个检查list1里的元素，同时跟踪list2中各元素的剩余可匹配次数，精准保留list1中“多余”的重复元素：

from collections import Counter

list1 = ['E', 'I', 'N', 'A', 'O', 'N', 'R']
list2 = ['I', 'A', 'N', 'R']

# 先统计list2中各元素的出现次数
list2_counts = Counter(list2)
difference = []

for item in list1:
    # 如果当前元素在list2还有剩余次数，就消耗一次，不加入差集
    if list2_counts.get(item, 0) > 0:
        list2_counts[item] -= 1
    # 否则，加入差集
    else:
        difference.append(item)

print(difference)  # 输出: ['E', 'O', 'N']

为什么这个方法有效？ #

• 我们用Counter记录list2里每个元素出现的次数，比如'N'出现1次。
• 遍历list1时，遇到第一个'N'，我们消耗list2里的那个'N'计数（减到0）；遇到第二个'N'时，list2里的'N'计数已经为0，所以这个'N'就会被加入差集。
• 完全保留了list1的原始顺序，和你想要的结果完全一致。

方法2：基于Counter的统计展开法 #

如果你不关心顺序，只需要得到正确的元素和次数，可以用Counter的减法操作，再把结果展开成列表：

from collections import Counter

list1 = ['E', 'I', 'N', 'A', 'O', 'N', 'R']
list2 = ['I', 'A', 'N', 'R']

count1 = Counter(list1)
count2 = Counter(list2)

# Counter的减法会自动保留正的计数，负数会被忽略
diff_counter = count1 - count2

# 展开成列表
difference = []
for item, count in diff_counter.items():
    difference.extend([item] * count)

print(difference)  # Python3.7+会输出 ['E', 'O', 'N']（保持插入顺序）

注意点： #

• Python3.7及以上版本中，Counter会保留元素的插入顺序，所以展开后的列表顺序和list1一致；如果是更早的版本，顺序可能不确定。

内容的提问来源于stack exchange，提问作者Facky Packy