關於 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)

希望它以某種方式有所幫助。