我在 Python 中有两个列表,如下所示:
temp1 = ['One', 'Two', 'Three', 'Four']
temp2 = ['One', 'Two']
我需要用第一个列表中的项目创建第二个列表,而第二个列表中没有这些项目。从示例中,我必须得到:
temp3 = ['Three', 'Four']
有没有循环和检查的快速方法吗?
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]))
。
现有解决方案均提供以下一项或多项:
但是到目前为止,还没有解决方案。如果两者都想要,请尝试以下操作:
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
temp3 = [item for item in temp1 if item not in temp2]