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)