繁体   English   中英

python中字典值的加权平均值

[英]Weighted average of dictionary values in python

我有一个由234个词典组成的参数网格,每个词典都有相同的键。 然后,我有一个权重列表,通过这些权重列表,我可以计算出这些字典值的加权平均值。 换句话说,我只需要处理一个字典,它具有与初始243相同的键,但是作为附加到每个键上的值,需要使用243权重进行加权平均。

我尝试使用Counter来累积结果,但是它返回的值非常小,对我来说没有意义。 w[0]是243个权重的列表,每个权重与grid内的243个词典有关

from collections import Counter

def avg_grid(grid, w, labels=["V0", "omega", "kappa", "rho", "theta"]):

    avgHeston = Counter({label: 0 for label in labels})

    for i in range(len(grid)):

        avgHeston.update(Counter({label: grid[i][label] * w[0][i] for label in labels}))

    totPar = dict(avgHeston)

    return totPar

有没有更容易实现的方法?

您可能要改用defaultdict

from collections import defaultdict

def avg_grid(grid, wgts, labels=["V0", "rho"]):

    avgHeston = defaultdict(int)

    for g,w in zip(grid, wgts):
        for label in labels:
            avgHeston[label] += g[label] * w

    return avgHeston

weights = [4,3]
grd = [{'V0':4,'rho':3}, {'V0':1,'rho':2}]

print(avg_grid(grd, weights))

输出:

defaultdict(<class 'int'>, {'V0': 19, 'rho': 18})

笔记:

我更改了w以便您需要直接输入清单。 您可能想要这样调用函数: avg_grid(grids, w[0])

同样,这不会产生平均值。 您可能需要在某个时候除以len(grid)

for g,w in zip(grid, wgts):是更具Python风格的迭代

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM