Python中根据属性值从对象列表获取对应对象引用的最佳方法

你现在用的两种写法都能实现功能，但都有明显的优化空间，短板也十分突出：

• 写法1：[i for i in myList if i.attr==2][0]
  • 可取之处：逻辑直白，读代码的人一眼就能懂是筛选符合条件的元素取第一个，只需要单轮遍历原列表
  • 存在的问题：因为你明确知道实例的attr值是唯一的，找到第一个匹配项之后完全没必要继续遍历剩下的元素，但列表推导的机制是必须跑完所有元素、把所有匹配结果收集成新列表之后，才会取下标0的元素；如果列表长度很长、目标元素在靠前的位置，就会做大量无意义的遍历。另外如果没有任何匹配项，直接取[0]会抛出IndexError: list index out of range，报错信息完全没提到是attr匹配的问题，排查起来要翻半天代码。
• 写法2：myList[[i.attr for i in myList].index(2)]
  • 这种写法基本没有可取之处，属于能不用就不用的方案：首先要完整遍历整个列表，生成一个和原列表等长的临时列表存所有attr值，之后再在这个临时列表上找目标值的索引，相当于做了两次全列表遍历，还多占了一份和原列表等大的临时内存。在列表规模大的时候，这个写法的性能比第一种差至少一倍，如果目标值不存在，index()抛出的ValueError同样没有业务上下文，排查成本很高。

如果你的查找是低频操作，列表规模在万级及以下，不需要做复杂的结构改造，用Python内置的惰性生成器配合next的写法，就比你现在用的两种方案都好：

# 找到第一个匹配对象就立刻终止遍历，没有额外的列表内存开销
target = next(obj for obj in myList if obj.attr == 2)

如果担心找不到匹配项直接抛异常，可以给next加个兜底的默认返回值：

# 找不到就返回None，不会触发异常
target = next((obj for obj in myList if obj.attr == 2), None)

这个写法的核心优势：

• 生成器是边遍历边判断，找到匹配项立刻停止执行，不会像列表推导那样强制跑完全表，遍历效率比两种原方案更高
• 过程中不会生成额外的临时列表，没有多余的内存开销
• 可以灵活自定义异常逻辑，如果业务要求必须找到匹配对象，就加个捕获逻辑，抛出带明确上下文的错误，排查起来方便很多：

  try:
      target = next(obj for obj in myList if obj.attr == 2)
  except StopIteration:
      raise ValueError("未找到attr值为2的对象")
  

如果你的列表规模到了十万级以上，而且需要非常频繁的按这个attr值查找对象，那不管怎么优化遍历写法，每次查找都是O(n)的时间复杂度，规模上去之后性能会非常拉胯。这时候最优方案是一次性把列表转成attr到对象的映射字典，把后续的查找复杂度降到O(1)：

# 仅需一次全量遍历就能构建好映射关系
obj_attr_map = {obj.attr: obj for obj in myList}

# 之后每次查找直接按键取，瞬时完成，没有遍历开销
target = obj_attr_map.get(2)

这个方案的注意事项：

• 如果列表是静态的（不会频繁增删元素、也不会修改实例的attr值），这个方案的性能是碾压所有遍历写法的，列表规模越大、查找次数越多，收益越明显
• 如果列表是动态变化的，需要同步更新字典映射，避免出现字典存储的内容和实际列表不一致的问题

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