基于Python中的其他列表，列表理解的过滤嵌套列表

Question

我有两个这样的列表：

a = [[1,2,3],[2,3,4],[5,6,7],[7,8,9]]
b = [1,2]

我现在想过滤列表a，以保留列表b中包含项目的项目。因此，所需的输出看起来像这样：

[[1,2,3],[2,3,4]]

我尝试了一些嵌套的列表综合，我可以想到，但无法获得所需的输出。任何建议将不胜感激。

Answer 1

您可以尝试这样的事情：

print([i for i in a if any(map(i.__contains__,b))])

>>> [[1, 2, 3], [2, 3, 4]]

Answer 2

我会尝试这样的事情：

a = [[1,2,3],[2,3,4],[5,6,7],[7,8,9]]
b = [1,2]

result = [lst for lst in a if any(x in lst for x in b)]

Answer 3

列表理解和集合的结合将产生所需的结果。笔记;我认为，如果情况是这样，重复的项目和订购就不感兴趣 - 一组将无法正常工作，因为它忽略了排序，并且只允许唯一的项目。

一个简单的列表理解将会做到，就像下面的

filter_items = set(filter_items)
[sublist for sublist in original_list if not set(sublist).isdisjoint(filter_items)]

主要是此列表理解的一个有趣的部分，即（如果未设置）（sublist）.isdisjoint（filter_items）。在这里，只有当sublist的集合不是 dissem_items ie ie ie ie ie ie ie ie ie ie ie ie ie ie ie ie n none none none，才能保留sublist。 filter_items的属于sublist。

对于您给定的示例，提供的答案将产生以下内容：

>>> a = [[1,2,3],[2,3,4],[5,6,7],[7,8,9]]
>>> b = set([1,2])
>>> [sublist for sublist in a if not set(sublist).isdisjoint(b)]
[[1, 2, 3], [2, 3, 4]]

Answer 4

使用set方法中的可以模仿相交：

a = [[1,2,3],[2,3,4],[5,6,7],[7,8,9]]
b = [1,2]

b_as_set = set(b)

out = [l for l in a if b_as_set.intersection(l)]
# [[1, 2, 3], [2, 3, 4]]

Answer 5

我会尝试这样的事情。

a = [[1,2,3],[2,3,4],[5,6,7],[7,8,9]]
b = [1,2]

print([lst for lst in a if any([item in b for item in lst])])