Быстрый вопрос о пересечении списков в Python

Question

Я очень новичок в python, и у меня есть быстрый вопрос о сравнении списков. У меня очень простой код. Я создал массив, используя и хотел пересечение второго обозначенного элемента массива и определенного списка, который у меня есть. Этот список постоянно изменяется в используемых тегах, но он состоит из цифр. Во всяком случае, из этого я получаю пересечение, которое я хочу homever, я хочу сделать еще одну вещь с этим кодом: после того, как я использую функцию пересечения, я хочу получить 0-й и 1-й элементы строки. Т. е. Print (пересечение (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)

Надеюсь, это поможет каким-то образом.