Получение нужных данных в нужном месте

У меня есть файл Json, я смог извлечь из него некоторые значения и просуммировать их. Я хочу иметь возможность поместить результат в нужный ключ, но не могу понять, как это сделать. Ниже приведен мой код:

from builtins import print
import json
import jmespath
from collections import Counter

const = ['constituency A','constituency B','constituency C']
region = ['region A','region B','reigon C']
poll = ['POLLING STATION A','POLLING STATION B','POLLING STATION C','POLLING STATION   Z','POLLING STATION F']

fake = {'transaction':[{'region A':{'constituency A':{
    'POLLING STATION A':{'PARTY A':10,'PARTY B':20,'PARTY C':30,'PARTY D':40},
    'POLLING STATION Z':{'PARTY A':50,'PARTY B':60,'PARTY C':70,'PARTY D':80},
    'POLLING STATION B':{'PARTY A':90,'PARTY B':100,'PARTY C':110,'PARTY D':120},
    'POLLING STATION F':{'PARTY A':190,'PARTY B':1100,'PARTY C':1110,'PARTY D':1120},},
    }}]}

 a = json.dumps((fake))
 p = json.loads(a)
 j = jmespath.search('transaction[*]',p)

 ham = []
 man = set()
 for new_d in j:
     for k,v in new_d.items():
         for i_k,i_v in v.items():
             for w,c in i_v.items():
                 if w in poll and i_k in const and k in region:
                     ham.append(c)
 up = len(ham)
 i= 0
 a1=Counter()
 while i < up:
     a1 += Counter(ham[i])
     i+=1

  print(a1)

Вот что я хотел сделать, результат, который является a1, будет помещен в словарь таким образом =>[ {'region A':{'constituency A':{'PARTY D': 1360, 'PARTY C': 1320, 'PARTY B': 1280, 'PARTY A': 340}}}]. Когда голоса за избирательный округ B в регионе A также подсчитаны, результат будет добавлен к региону A с избирательным округом B в качестве ключа.

Я провел итерации по каждому дикту и подсчитал голоса партий для каждого избирательного округа.

fake = {
    'transaction': [
        { 'region A': 
            { 'constituency A': 
                {
                    'POLLING STATION A': 
                        {'PARTY A': 10, 'PARTY B': 20, 'PARTY C': 30, 'PARTY D': 40},
                    'POLLING STATION Z': 
                        {'PARTY A': 50, 'PARTY B': 60, 'PARTY C': 70, 'PARTY D': 80},
                    'POLLING STATION B': 
                        {'PARTY A': 90, 'PARTY B': 100, 'PARTY C': 110, 'PARTY D': 120},
                    'POLLING STATION F': 
                        {'PARTY A': 190, 'PARTY B': 1100, 'PARTY C': 1110, 'PARTY D': 1120},
                },
            }
        }
    ]
}

data = fake['transaction'][0]

total_result = {}

for region, constituencies in data.items():
    total_result[region] = {}
    for constituency, stations in constituencies.items():
        party_votes = {}
        for station, parties in stations.items():
            for party, vote in parties.items():
                party_votes[party] = party_votes.get(party, 0) + vote
        total_result[region][constituency] = party_votes

print(total_result)

печатает

{'region A': {'constituency A': {'PARTY A': 340, 'PARTY B': 1280, 'PARTY C': 1320, 'PARTY D': 1360}}}
Вернуться на верх