Python如何高效判断指定元素是否存在于元组列表中

1. 现有方案的效率评估 #

你当前使用的遍历比对方案时间复杂度为O(n)，n为categories的元素长度：

• 如果categories的元素规模很小（几十到上百个），该方案完全够用，性能不会有明显瓶颈
• 如果categories元素过万、或需要高频重复查询，这个方案的效率就不足以支撑需求了

2. 基于lambda的实现方式 #

可以搭配any()函数+lambda实现单行判断，代码会更简洁：

is_exist = any(map(lambda x: x[0] == user_category_input, categories))

还有更常用的简化写法，不需要lambda也能实现相同逻辑，效率和遍历完全一致：

is_exist = any(category[0] == user_category_input for category in categories)

3. 实际开发的优化方案 #

根据使用场景选择对应优化方案即可：

• 单次/低频查询、数据量小：直接用上面的any()写法就行，代码简洁易读，不需要额外改造
• 高频查询、数据量大：提前把所有分类名提取出来存入集合，后续查询的时间复杂度可以降到O(1)：

  # 初始化阶段仅执行一次
  category_names = {cat[0] for cat in categories}
  # 后续每次查询直接判断
  is_exist = user_category_input in category_names
  

• 如果业务本身就需要频繁通过分类名获取对应数值，直接把categories转成字典是最优选择：

  # 初始化阶段仅执行一次
  category_map = dict(categories)
  # 既可以判断分类是否存在，也可以直接获取对应数值
  is_exist = user_category_input in category_map
  if is_exist:
      category_value = category_map[user_category_input]
  

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