python - 在字典列表中按类别获取最大值

Question

我有一个看起来像这样的字典列表:

example = [
    {'id': 1, 'foo': 10, 'seq': 1, 'val':  0, 'sum':  0} ,
    {'id': 1, 'foo': 94, 'seq': 2, 'val':  5, 'sum':  5} ,
    {'id': 1, 'foo': 32, 'seq': 3, 'val': 32, 'sum': 37} ,
    {'id': 2, 'foo': 10, 'seq': 1, 'val':  0, 'sum':  0} ,
    {'id': 2, 'foo': 43, 'seq': 2, 'val':  3, 'sum':  3} ,
    {'id': 2, 'foo': 71, 'seq': 3, 'val': 14, 'sum': 17} ,
    {'id': 2, 'foo': 32, 'seq': 4, 'val':  6, 'sum': 23} ,
    {'id': 3, 'foo': 26, 'seq': 1, 'val':  0, 'sum':  0} ,
    {'id': 3, 'foo': 38, 'seq': 2, 'val':  8, 'sum':  8}
]

我需要列表中的三个关键信息:

1) 我需要为列表中的每个“id”确定最高的“seq”值。

2) 使用 (1) 的结果，我需要将每个最高“seq”值的“sum”值与其他最高“seq”值进行比较，并确定哪个“seq”值具有最低的“sum” .

3) 这个过程很复杂，因为我还需要根据 'foo' 值进行比较，这样 (2) 中的比较只比较第一个和最后一个 'foo' 的 'sum' 值值相同。

因此，在示例列表中，我希望将“id”值 1 和 2 相互比较，因为它们以“foo”10 开头并以“foo”32 结尾。

这里有一些伪代码，我希望它能实现我想要做的事情:

def getMinId(foo1, foo2, exampleList):
    # first limit exampleList to only ids that match the foos
    # next find the minimum 'sum' among the remaining ids

所需的最终输出可能类似于:

{(10,32): 23, (26,38): 8}

我可以通过对列表进行大量迭代来强制执行此操作，但我想知道是否有更有效的方法来做到这一点。唯一真正重要的结果是找到每个“foo”对的最小“总和”值，因此如果有更好的思考方式，请忽略我在此处列出的步骤。

Answer 1

这将适用于您的示例(在 Python 2.7 中)。请务必在更大的数据集上对其进行测试。

ma​​xvaldict.py

example = [
    {'id': 1, 'foo': 10, 'seq': 1, 'val':  0, 'sum':  0},
    {'id': 1, 'foo': 94, 'seq': 2, 'val':  5, 'sum':  5},
    {'id': 1, 'foo': 32, 'seq': 3, 'val': 32, 'sum': 37},
    {'id': 2, 'foo': 10, 'seq': 1, 'val':  0, 'sum':  0},
    {'id': 2, 'foo': 43, 'seq': 2, 'val':  3, 'sum':  3},
    {'id': 2, 'foo': 71, 'seq': 3, 'val': 14, 'sum': 17},
    {'id': 2, 'foo': 32, 'seq': 4, 'val':  6, 'sum': 23},
    {'id': 3, 'foo': 26, 'seq': 1, 'val':  0, 'sum':  0},
    {'id': 3, 'foo': 38, 'seq': 2, 'val':  8, 'sum':  8}
]

# Dictionary identifying highest 'seq' value for each 'id' in the list.
# {id: [seq, sum, first_foo, last_foo]}
maxiddict = {}
for rowdict in example:
    if not maxiddict.get(rowdict['id']) or rowdict['seq'] > maxiddict.get(rowdict['id'])[0]:
        if not maxiddict.get(rowdict['id']):
            maxiddict[rowdict['id']] = [rowdict['seq'], rowdict['sum'], rowdict['foo'], rowdict['foo']]
        else:
            first_foo = maxiddict[rowdict['id']][2]
            maxiddict[rowdict['id']] = [rowdict['seq'], rowdict['sum'], first_foo, rowdict['foo']]

# Dictionary of groups of maxiddicts grouped by a combined key of first_foo, last_foo
# {'first_foo, last_foo': [{id: [seq, sum, first_foo, last_foo]}]}
groupdict = {}
for k, v in maxiddict.items():
    key = ', '.join([str(v[2]), str(v[3])])
    if not groupdict.get(key):
        groupdict[key] = [dict([(k, v)])]
    else:
        groupdict[key].append(dict([(k, v)]))

# Dictionary of lowest sums of maxidicts entries grouped by combined key of first_foo, last_foo
lowestsumsdict = {}
for groupkey, groupvallist in groupdict.items():
    minsum = min([entry.itervalues().next()[1] for entry in groupvallist])
    lowestsumsdict[groupkey] = minsum

print lowestsumsdict

输出:

(maxvaldict)macbook:maxvaldict joeyoung$ python maxvaldict.py 
{'26, 38': 8, '10, 32': 23}