Python: lista filtrów listy z inną listą

Question

Próbuję filtrować listę, chcę wyodrębnić z listy A (jest listą list), elementy, które pasują do klucza indeks 0, z inną listą B co ma serie wartości

jak ten

list_a = list(
    list(1, ...), 
    list(5, ...), 
    list(8, ...), 
    list(14, ...) 
) 

list_b = list(5, 8) 

return filter(lambda list_a: list_a[0] in list_b, list_a) 

powinien zwrócić:

list(
    list(5, ...), 
    list(8, ...) 
) 

Jak mogę to zrobić? Dzięki!

Answer 1

Użyj listowych:

result = [x for x in list_a if x[0] in list_b] 

celu zwiększenia wydajności konwersji list_b do zestawu pierwszej kolejności.

Jako że @kevin zanotował w komentarzach coś takiego jak list(5,8) (chyba że nie jest to pseudo-kod) jest nieprawidłowe, a otrzymasz komunikat o błędzie.

list() akceptuje tylko jeden element i że element powinien być iterable/iterator

Answer 2

Jesteś rzeczywiście bardzo blisko. Po prostu wykonaj to:

list_a = list(
    list(1, ...), 
    list(5, ...), 
    list(8, ...), 
    list(14, ...) 
) 

# Fix the syntax here 
list_b = [5, 8] 

return filter(lambda list_a: list_a[0] in list_b, list_a)