检查元素是否存在时获取子列表的索引

Question

作为这个问题的延续，我怎样才能获得包含该元素的子列表的索引？

例如，如果我有：

a = [[1,2],[3,4],[5,6]]
any(2 in i for i in a)
....

结果如何得到 0（因为 a[0] 包含数字 2）？

Answer 1

或者，你也可以试试这个：

for idx, sublst in enumerate(a):
    if 2 in sublst: 
        print(idx)

Output：

0

Answer 2

使用enumerate()对带有索引的列表进行迭代的简单列表推导可以解决问题：

res = [i for i, el in enumerate(a) if 2 in el]

您可以使用常规for循环实现相同的目的：

res = []
for i, el in enumerate(a):
    if 2 in el:
        res.append(el)