python - Python：用另一个列表过滤列表列表

Question

我正在尝试过滤列表，我想从列表 A（是列表列表）中提取与键索引 0 匹配的元素，另一个列表 B 具有一系列值

像这样

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)

应该返回：

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

我怎样才能做到这一点？谢谢！

Answer 1

使用列表推导：

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

为了提高性能list_b，请先转换为一组。

正如@kevin 在评论中指出的那样list(5,8)（除非它不是伪代码）是无效的，你会得到一个错误。

list()只接受一个项目，该项目应该是可迭代的/迭代器

Answer 2

你实际上非常接近。只需这样做：

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)