在字典列表

时间:2015-10-10 21:21:11

标签: python python-2.7

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

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)中的比较只比较'sum'值,其中第一个和最后一个'foo'价值是一样的。

因此,在示例列表中,我希望将'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'对的最小'sum'值,所以如果有更好的方法来考虑它,请随意忽略我在这里列出的步骤。

1 个答案:

答案 0 :(得分:0)

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

<强> maxvaldict.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}