获取两个列表之间的差异

2022-09-05 01:04:39

我在Python中有两个列表:

temp1 = ['One', 'Two', 'Three', 'Four']
temp2 = ['One', 'Two']

我想创建第三个列表,其中包含第一个列表中不在第二个列表中的项目:

temp3 = ['Three', 'Four']

有没有没有周期和检查的快速方法?


答案 1

要获取位于 中但不在 中的元素,temp1temp2

In [5]: list(set(temp1) - set(temp2))
Out[5]: ['Four', 'Three']

请注意它是不对称的:

In [5]: set([1, 2]) - set([2, 3])
Out[5]: set([1]) 

您可能希望/希望它等于 .如果您确实想要作为答案,则可以使用.set([1, 3])set([1, 3])set([1, 2]).symmetric_difference(set([2, 3]))


答案 2

现有的解决方案都提供以下一种或另一种:

  • 比 O(n*m) 性能更快。
  • 保留输入列表的顺序。

但到目前为止,还没有解决方案同时具备这两种解决方案。如果您两者都想要,请尝试以下操作:

s = set(temp2)
temp3 = [x for x in temp1 if x not in s]

性能测试

import timeit
init = 'temp1 = list(range(100)); temp2 = [i * 2 for i in range(50)]'
print timeit.timeit('list(set(temp1) - set(temp2))', init, number = 100000)
print timeit.timeit('s = set(temp2);[x for x in temp1 if x not in s]', init, number = 100000)
print timeit.timeit('[item for item in temp1 if item not in temp2]', init, number = 100000)

结果:

4.34620224079 # ars' answer
4.2770634955  # This answer
30.7715615392 # matt b's answer

我提出的方法以及保持顺序也(略微)比集合减法快,因为它不需要构造不必要的集合。如果第一个列表比第二个列表长得多,并且哈希很昂贵,则性能差异将更加明显。这是第二个测试来证明这一点:

init = '''
temp1 = [str(i) for i in range(100000)]
temp2 = [str(i * 2) for i in range(50)]
'''

结果:

11.3836875916 # ars' answer
3.63890368748 # this answer (3 times faster!)
37.7445402279 # matt b's answer

推荐