如何使用python求和列值

2024-05-14 12:53:45 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个如下所示的行集:

defaultdict(<type 'dict'>, 
{
   u'row1': {u'column1': 33, u'column2': 55, u'column3': 23}, 
   u'row2': {u'column1': 32, u'column2': 32, u'column3': 17}, 
   u'row3': {u'column1': 31, u'column2': 87, u'column3': 18}
})

我希望能很容易地得到第1列,第2列,第3列的和。如果我可以对任意数量的列执行此操作,并在一个看起来像columnName => columnSum的哈希映射中接收结果,那就太好了。正如您可能猜到的那样,我不可能首先从数据库中获取汇总值,因此有理由提出这个问题。在


Tags: 数据库数量typedict汇总row1row2row3
2条回答

不是最棒的Python,但这里:

for val in defaultdict.values() :
   sum1 += val['column1']
   sum2 += val['column2']
   sum3 += val['column3']
final_dict = {'column1' : sum1,'column2' : sum2,'column3' : sum3 }

嵌套生成器+列表理解可以做到:

>>> foo
defaultdict(<type 'dict'>, {u'row1': {u'column1': 33, u'column3': 23, u'column2': 55}, u'row2': {u'column1': 32, u'column3': 17, u'column2': 32}, u'row3': {u'column1': 31, u'column3': 18, u'column2': 87}})
>>> dict(zip(foo.values()[0].keys(), [sum(j[k] for j in (i.values() for _,i in foo.items())) for k in range(3)]))
{u'column1': 96, u'column3': 58, u'column2': 174}

相关问题 更多 >

    热门问题