如果某个字典中的某个键具有相同的月份，该如何计算字典中的值？

Question

我有Django queryset：

summary = Contracts.objects.all() \
                   .filter(datestart__gte=time_threshold) \
                   .values('datestart', 'id_deposits__id_valuta__name') \
                   .annotate(sum=Sum('suma')) \
                   .order_by('-id_deposits__id_valuta__name')

我通过以下方式捕捉到qs：

for s in summary:
    for k, v in CURRENCY.items():
        if s['id_deposits__id_valuta__name'] == k:
            s['sum'] *= CURRENCY[k]

它会根据需要更新货币金额。 它返回字典

{'sum': Decimal('400000.00'), 'datestart': datetime.date(2016, 3, 1), 'id_deposits__id_valuta__name': 'Гривня'}
{'sum': Decimal('2500000.00'), 'datestart': datetime.date(2016, 4, 25), 'id_deposits__id_valuta__name': 'Долар'}
{'sum': Decimal('2500000.00'), 'datestart': datetime.date(2016, 4, 25), 'id_deposits__id_valuta__name': 'Долар'}

现在，当“ datestart”值中的月份相等时，我需要对字典中“ sum”键的值求和。

防爆。 有些字典有sum，datestart键。 当字典在同一个月时，我需要具有“和”值的总和：2016-04-03，2016-04-14

那么如何以简约的方式做到这一点呢？

Answer 1

您可以将每个字典转换为以月为键且值为s['sum'] Counter ，然后将它们加在一起：

from collections import Counter

sum((Counter({x['datestart'].month: x['sum']}) for x in summary), Counter()) 
# Counter({4: Decimal('5000000.00'), 3: Decimal('400000.00')})

请注意，如果需要区分月份和年份，请改用{(x['datestart'].month, x['datestart'].year): x['sum']} 。

Answer 2

使用字典来跟踪每月的总和； 您可以在此处使用defaultdict来管理初始值：

from collections import defaultdict
from decimal import Decimal

per_month = defaultdict(Decimal)

for s in summary:
    s['sum'] *= CURRENCY.get(s['id_deposits__id_valuta__name'], 1)
    month = s['datestart'].year, s['datestart'].month
    per_month[month] += s['sum']

我已经通过CURRENCY.items()删除了您的循环； 您可以只使用dict.get()来获取值； 如果键不在词典中，则使用1代替。

请注意，假设您希望将不同年份中的月份作为单独的总和，我将您的总和分组为(year, month)元组。