python - 通过匹配列表的字符串列表获取列表索引

Question

p = [['a','b','c'],['x','y','z']]

如何获取包含字符串(char)'a'的列表索引？

列表中的“a”不起作用，或者至少我无法弄清楚。

Answer 1

使用in，但在子列表中，而不是在列表的全局列表中。与enumerate结合获取匹配列表的索引

p = [['a','b','c'],['x','y','z']]

idx = [i for i,sublist in enumerate(p) if 'a' in sublist]

这给你一个索引列表。这里我们只有一个元素:

>>> idx
[0]

您可以使用相同的技术来计算子列表中 a 的索引 BTW(使用 idx = [i for i,sublist in enumerate(p) if 'a' in sublist ])

要获得唯一值或None，只需在生成器理解上迭代一次即可:

idx = next((i for i,sublist in enumerate(p) if 'a' in sublist),None)