程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
您现在的位置: 程式師世界 >> 編程語言 >  >> 更多編程語言 >> Python

python 兩個list 求交集,並集,差集

編輯:Python

大家好,又見面了,我是你們的朋友全棧君。

項目github地址:bitcarmanlee easy-algorithm-interview-and-practice 經常有同學私信或留言詢問相關問題,V號bitcarmanlee。github上star的同學,在我能力與時間允許范圍內,盡可能幫大家解答相關問題,一起進步。

在python中,數組可以用list來表示。如果有兩個數組,分別要求交集,並集與差集,怎麼實現比較方便呢? 當然最容易想到的是對兩個數組做循環,即寫兩個for循環來實現。這種寫法大部分同學應該都會,而且也沒有太多的技術含量,本博主就不解釋了。這裡給大家使用更為裝bility的一些方法。

老規矩,talk is cheap,show me the code

#!/usr/bin/env python
#coding:utf-8
'''
Created on 2016年6月9日
@author: lei.wang
'''
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()

讓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.使用列表解析式。列表解析式一般來說比循環更快,而且更pythonic顯得更牛逼。 2.將list轉成set以後,使用set的各種方法去處理。

發布者:全棧程序員棧長,轉載請注明出處:https://javaforall.cn/151880.html原文鏈接:https://javaforall.cn


  1. 上一篇文章:
  2. 下一篇文章:
Copyright © 程式師世界 All Rights Reserved