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

Question

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

例如，如果我有：

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

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

标签： python

Answer 1

使用索引遍历列表的简单列表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)