根据多个条件将列表拆分为多个子列表的最快方法是什么？

Question

根据条件将列表拆分为多个子列表的最快方法是什么？每个条件代表一个单独的子列表。

一种将listOfObjects 拆分为子列表的方法（三个子列表用于演示，但也可以更多）：

listOfObjects = [.......]
l1, l2, l3 = [], [], []
for l in listOfObjects:
    if l.someAttribute == "l1":
        l1.append(l)
    elif l.someAttribute == "l2":
        l2.append(l)
    else:
        l3.append(l)

这种方式看起来一点也不像pythonic，而且还需要相当长的时间。是否有更快的方法，例如使用map?

有一个类似的问题，但只有一个条件，即两个结果列表：How to split a list based on a condition?

Answer 1

您可以在此处collections.defaultdict 进行映射。

from collections import defaultdict

d = defaultdict(list)

for l in listOfObjects:
    d[l.someAttribute].append(l)

out = d.values() 
l1 , l2, l3 = d['l1'], d['l2'], d['l3']

d 的格式为。

{ 
  attr1 : [...],
  attr2 : [...],
  ...
  attrn : [...]
}

Answer 2

类似问题的答案令人惊叹。我没有考虑过拆分...无论如何，您可以执行类似的操作，但可读性会降低：

for l in listOfObjects:
    (l3, l2, l1)[(l.someAttribute == "l1")*2 or l.someAttribute == "l2"].append(l)

这适用于任何布尔条件。 or 返回第一个真值（或 False）。 True==1，所以我们为希望等于 2 的索引添加 *2。

但正如我所说，它不是真正可读的。而且不可扩展。

至于速度：or 是短路的，返回第一个真值，所以条件检查应该和你的方法类似。您可能希望将查找元组定义在循环之外。

---

而且使用 dict 更具可读性，因为您的条件基于相等（注意：您想要的属性也必须是可散列的）

lookup = {"l1": l1, "l2": l2}
for l in listOfObjects:
    lookup.get(l.someAttribute, l3).append(l)

dict.get 将默认值设为第二 - 所以它非常适合我们的 else 包罗万象。

在速度方面：字典查找将只有一次检查，而不是 or 链的条件链 ifs