如何在Python列表中基于函数求值直接查找元素索引？

我希望在Python列表中，基于lambda函数（或其他可调用对象）的求值结果查找元素的索引，相当于将.index()操作与find_if操作结合使用。当前我需要先找到匹配元素，再查找其对应的索引，能否将这两个操作合并为一步？

示例代码如下：

# self 包含 self.list
def find_index_where(self, id: int) -> int:

    # 用于测试对象子成员的可调用表达式
    def lambda_callable(input, id):
        return input.id == id

    # 当前的额外操作：
    # 先找到匹配元素，再查找该元素对应的索引
    matched_item = next(item for item in self.list if lambda_callable(item, id))
    index = self.list.index(matched_item)
    return index

当然可以合并成一步，核心是用enumerate同时遍历索引和元素，直接在遍历过程中判断条件并返回索引，避免二次查找。以下是几种实现方式：

直接内嵌条件判断 #

最简洁的写法，直接把判断逻辑写在生成器表达式里：

def find_index_where(self, id: int) -> int:
    return next(idx for idx, item in enumerate(self.list) if item.id == id)

保留可调用对象的写法 #

如果需要把判断逻辑抽成单独的函数（或lambda），可以这样写：

def find_index_where(self, id: int) -> int:
    # 单独的判断函数
    def matches(item):
        return item.id == id
    
    return next(idx for idx, item in enumerate(self.list) if matches(item))

# 或者用lambda简化
def find_index_where(self, id: int) -> int:
    matcher = lambda item: item.id == id
    return next(idx for idx, item in enumerate(self.list) if matcher(item))

处理无匹配项的情况 #

如果担心列表中没有符合条件的元素会抛出StopIteration异常，可以给next()指定默认值（比如返回-1）：

def find_index_where(self, id: int) -> int:
    return next((idx for idx, item in enumerate(self.list) if item.id == id), -1)

这种方式的优势在于：

• 只需要遍历一次列表，比先找元素再查索引的二次遍历效率更高
• 代码更简洁，逻辑连贯，不需要额外存储匹配元素

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