python - 如何检查Python中的列表列表中是否存在某个元素？

Question

我有一个与文件中的行相对应的列表列表，具有多列。

[[col1, col2, col3], [elem1, elem2, elem3], [elem4, elem5, elem6]] 

我想检查(例如)elem3 是否在任何列表中，如果是，请进入该列表。 (实际上我有一个需要检查的内容列表，所以它可能包含 elem3、elem5、elem7....等)

Answer 1

最短的方法是使用列表推导式，列表推导式有时比简单的for循环更快

您的列表:

list1 =  [ ["col1", "col2", "col3"], ["elem1", "elem2", "elem3"], ["elem4", "elem5", "elem6"] ]

要查找的元素:

to_find = "col1"

你的函数“进入该列表”:

def do_something(sub_list):
    print (sub_list)

和列表理解将找到您的元素并使用包含它的列表调用函数:

[do_something(sub_list) for sub_list in list1 if to_find in sub_list]