汇总Python对象列表中的属性

Question

类似于“ Python中用同一属性对对象列表进行分组和汇总的最简洁方法是什么” ，我有一个脚本，需要在其中添加对象列表的属性。 但是，我的问题略有不同。

我有一类具有属性V，W，X，Y和Z的对象。我需要通过遍历属性W，X和Y并将其与所有其他W，X和Y属性进行匹配来对属性Z求和相同。 产生一个由W，X和Y索引的新求和值。

这是对象的类：

class xb(object):
    def __init__(self, V, W, X, Y, Z):
       self.V = V
       self.W = W
       self.X = X
       self.Y = Y
       self.Z = Z

xbs = [xb()]

我最初的想法是通过一系列嵌套的if语句来执行此操作，但是这会大大减慢处理速度，而且我确信我的逻辑是不明智的。

for xb in xbs:
    if xb.W == xb.W:
          if xb.X == xb.X:
              if xb.Y == xb.Y:
                  sum(xb.Z)

任何建议对此将不胜感激！

Answer 1

您可以使用defaultdict执行此操作：

from collections import defaultdict
indexed_sums = defaultdict(int)
for o in xbs:
    indexed_sums[(o.W, o.X, o.Y)] += o.Z

例如，如果您开始使用（使用xb的类定义）：

xbs = [xb(1, 2, 3, 4, 5),
       xb(1, 2, 3, 4, 5),
       xb(1, 2, 3, 4, 5),
       xb(1, 4, 3, 4, 5),
       xb(1, 4, 3, 4, 3),
       xb(1, 2, 3, 9, 3)]

您最终得到：

print dict(indexed_sums)
# {(4, 3, 4): 8, (2, 3, 4): 15, (2, 3, 9): 3}

因此，您可以获得W，X，Y为2、3、4的总和为：

indexed_sums[(2, 3, 4)]
# 15

---

请注意， defaultdict在这里所做的工作很少（这只是一个默认为0的计数字典）：主要是要在字典中索引(oW, oX, oY)元组。 如果没有defaultdict ，您可能会做同样的事情：

indexed_sums = {}
for o in xbs:
    if (o.W, o.X, o.Y) not in indexed_sums:
        indexed_sums[(o.W, o.X, o.Y)] = 0
    indexed_sums[(o.W, o.X, o.Y)] += o.Z

defaultdict只为您节省了两行。

Answer 2

这是一个单线的非常肮脏的技巧：

{key:sum(g.Z for g in group)
    for key, group in 
    itertools.groupby(
        sorted(L, key=lambda p:tuple((operator.attrgetter(a)(p) for a in 'VWXYZ'))),
        key=lambda p:tuple(
                      (operator.attrgetter(a)(p) for a in 'VWXYZ')
                     )
    )
}

我完全不建议这样做（调试起来很痛苦），但是我仍然认为这是一个有趣的解决方案