wutao1935

导航

python 两个list 求交集,并集,差集

def diff(listA,listB): #求交集的两种方式 retA = [i for i in listA if i in listB] retB = list(set(listA).intersection(set(listB))) print "retA is: ",retA print "retB is: ",retB #求并集 retC = list(set(listA).union(set(listB))) print "retC1 is: ",retC #求差集,在B中但不在A中 retD = list(set(listB).difference(set(listA))) print "retD is: ",retD retE = [i for i in listB if i not in listA] print "retE is: ",retE def main(): listA = [1,2,3,4,5] listB = [3,4,5,6,7] diff(listA,listB) if __name__ == '__main__': main() 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 让code run起来 retA is: [3, 4, 5] retB is: [3, 4, 5] retC1 is: [1, 2, 3, 4, 5, 6, 7] retD is: [6, 7] retE is: [6, 7] 1 2 3 4 5 结合代码来看,大体上是两种思路: 1.使用列表解析式。列表解析式一般来说比循环更快,而且更pythonic显得更牛逼。 2.将list转成set以后,使用set的各种方法去处理。

posted on 2018-01-20 18:23  逐风浪子  阅读(642)  评论(0编辑  收藏  举报