python - 关于 python 中列表交集的快速问题

Question

我对Python很陌生，我有一个关于比较列表的快速问题。我的代码非常简单。我使用创建了一个数组，并想要数组的第二个索引元素和我拥有的某个列表的交集。该列表在所使用的标签上不断变化，但它由数字组成。不管怎样，从这里我得到了我想要回家的交集，我想对此代码做另一件事:使用 intersect 函数后，我想获取该行的第 0 个和第 1 个元素。i.e print(intersection(D,liste[-1])) 结果是 [2,3] 那么我想得到 [40,75] 和 [75,25]这可能是一个非常简单的案例，但正如我所说，我是初学者并期待了解更多信息。提前致谢

A=np.array([[25,25,1],[40,75,2],[75,25,3],[25,50,10]])
C=[1,2,6]
B=[1,2,19]
D=[]
D.extend(A[:,2])
def intersection(D,B): 
    lst3 = [value for value in D if value in B] 
    return(lst3)
 print(intersection(D,liste[-1])) 

Answer 1

不确定我是否理解你的问题，但你可以将你的列表变成集合，以获得它们之间的交集，如下所示:

first_collection = set([1, 3, 4])
second_collection = set([1, 5, 7])

intersection = first_collection.insersection(second_collection)

希望它能有所帮助。