有沒有辦法讓我比較兩個不同的串列并只取出相似的串列？例子：
list1 = [(JOHN,BLUE,BANANA),(JOHN,BLUE, APPLE),(MARY,PURPLE,GRAPE),(BEN,GREEN,WATERMELON)
list2 = [(JOHN,BLUE),(MARY,PURPLE),(TOMMY,PINK)]
OUTPUT:
[(JOHN, BLUE, BANANA),(JOHN, BLUE, APPLE),(MARY, PURPLE, GRAPE)]
uj5u.com熱心網友回復：
只需檢查 list2 中 list1 中的 item[:2]
list1 = [
('JOHN', 'BLUE', 'BANANA'),
('JOHN', 'BLUE', 'APPLE'),
('MARY', 'PURPLE', 'GRAPE'),
('BEN', 'GREEN', 'WATERMELON'),
]
list2 = [
('JOHN', 'BLUE'),
('MARY', 'PURPLE'),
('TOMMY', 'PINK')]
print([item for item in list1 if item[:2] in list2])
[('JOHN', 'BLUE', 'BANANA'), ('JOHN', 'BLUE', 'APPLE'), ('MARY', 'PURPLE', 'GRAPE')]
uj5u.com熱心網友回復：
您可以嘗試以下基本方法：
list1 = [("JOHN","BLUE","BANANA"),("JOHN","BLUE", "APPLE"),("MARY","PURPLE","GRAPE"),("BEN","GREEN","WATERMELON")]
list2 = [("JOHN","BLUE"),("MARY","PURPLE"),("TOMMY","PINK")]
tmp = set()
for i in list1:
for j in list2:
if len(set(i)&set(j)) > 0:
tmp.add(i)
print tmp
結果：
set([('MARY', 'PURPLE', 'GRAPE'), ('JOHN', 'BLUE', 'BANANA'), ('JOHN', 'BLUE', 'APPLE')])
uj5u.com熱心網友回復：
def common_member(a, b):
a_set = set(a)
b_set = set(b)
if (a_set & b_set):
print(a_set & b_set)
else:
print("No common elements")
a = [JOHN,BLUE,BANANA),(JOHN,BLUE, APPLE),(MARY,PURPLE,GRAPE),(BEN,GREEN,WATERMELON]
b = [(JOHN,BLUE),(MARY,PURPLE),(TOMMY,PINK)]
common_member(a, b)
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/390777.html