Python - 如何查找列表（或子列表）中存在的项目？

Question

例如

my_list = [1, 2, 3, 4, [5, 6], [7, 8]]

我想找出 7 是否在my_list 。？ 答案应该是 True，因为它是最后一个子列表的一部分。 有任何想法吗？

Answer 1

如果mylist是只有列表的列表，您可以使用itertools.chain.from_iterable()

import itertools
mylist = [[1,2,3,4],[5,6],[7,8]]
merged = list(itertools.chain.from_iterable(mylist))
7 in merged

由于我们有一个与列表混合的整数列表，我们定义了一个自定义过滤器。 这也可以在单个列表理解中完成，但我现在无法弄清楚。

mylist = [1,2,3,4,[5,6],[7,8]]
merged = []
map(lambda l: merged.extend(l) if isinstance(l, list) else merged.append(l), [i for i in mylist])
7 in merged

您还可以使用已弃用的compiler.ast.flatten函数：

from compiler.ast import flatten
merged = flatten(mylist)