如何在嵌套列表中按子列表首元素查找其索引？

假设我们有嵌套列表 mylist = [[1,3],[2,2],[3,7]]，想要定位首个元素为3的子列表的索引（预期结果是2）。你原本想通过list.index()配合key参数来实现：

i = mylist.index(3, key = lambda x: x[0])

可惜list.index()确实不支持key参数，不过有几种更符合Python风格的实现方式，下面给你详细说：

推荐的Pythonic实现方法 #

1. 用enumerate() + next()生成器（最优解）

这是我最推荐的方式，因为生成器会在找到第一个匹配项后立刻停止遍历，不用处理整个列表，效率最高：

mylist = [[1,3],[2,2],[3,7]]
try:
    index = next(idx for idx, sublist in enumerate(mylist) if sublist[0] == 3)
    print(index)  # 输出: 2
except StopIteration:
    # 处理找不到匹配项的情况
    print("未找到首个元素为3的子列表")

这里用next()从生成器中取出第一个符合条件的索引，如果遍历完都没找到，会抛出StopIteration异常，所以记得用try-except捕获这种边界情况。

2. 列表推导式（适合需保留所有匹配索引的场景）

如果你之后可能需要用到所有符合条件的索引，或者不介意遍历整个列表，可以先收集所有匹配的索引，再取第一个：

mylist = [[1,3],[2,2],[3,7]]
matching_indices = [idx for idx, sublist in enumerate(mylist) if sublist[0] == 3]
if matching_indices:
    print(matching_indices[0])  # 输出: 2
else:
    print("未找到匹配项")

这种方式会先遍历整个列表生成所有匹配索引，再取第一个，在列表很大且匹配项靠前时，效率不如生成器，但胜在直观易懂。

3. 提取首元素列表后用index()（小列表场景适用）

虽然list.index()不支持key，但我们可以先把所有子列表的首个元素提取成一个新列表，再用index()找目标值的位置：

mylist = [[1,3],[2,2],[3,7]]
try:
    index = [sublist[0] for sublist in mylist].index(3)
    print(index)  # 输出: 2
except ValueError:
    print("未找到匹配项")

这种方式需要额外创建一个列表存储首元素，当原列表很大时会占用更多内存，所以只适合小列表的场景。

内容的提问来源于stack exchange，提问作者Zak Stucke