python--两个list 求交\并\差集

Python中,数组可以用list来表示。如果有两个数组,分别要求交集,并集与差集,怎么实现比较方便呢? 
当然最容易想到的是对两个数组做循环,即写两个for循环来实现(舍弃)。

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()

********************************************************

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.使用列表解析式。列表解析式一般来说比循环更快,而且更pythonic显得更牛逼。 

2.将list转成set以后,使用set的各种方法去处理。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值