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

lubię to

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 wrócić:

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

Jak mogę to zrobić? Dzięki!