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

Question

我对 python 非常陌生，我有一个关于比较列表的快速问题。我的代码非常简单。我使用数组的第二个指示元素和我拥有的某个列表的交集创建了一个数组。 该列表在使用的标签上不断变化，但由数字组成。 无论如何，从这里我得到了我想要回家的交集，我想对这段代码做另一件事：在我使用 intersect function 之后，我想获得行的第 0 和第 1 个元素。 İ.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)

希望它以某种方式有所帮助。