查找匹配的索引

查找给定列表中满足给定测试函数的第一个元素的索引。

  • 使用列表推导式、enumerate()next() 来返回满足 fn 返回 True 的第一个元素在 lst 中的索引。
def find_index(lst, fn):
  return next(i for i, x in enumerate(lst) if fn(x))

find_index([1, 2, 3, 4], lambda n: n % 2 == 1) # 0