import urllib.request
import json
dok=json.loads(urllib.request.urlopen("https://raw.githubusercontent.com/TartuNLP/grammar-worker/kaisall-patch-2/posgram_finder/model_postcontext.txt").read().decode("utf-8"))
#print(dok["AAA"]["$"])
uuritavad=["N", "K", "G", "Y", "X"]
loetelu={}
for u in uuritavad:
  loetelu[u]=[]
for kolmik in dok.keys():
  lisand={}
  for u in uuritavad: lisand[u]=0
  for ots in dok[kolmik].keys():
    if ots in uuritavad: lisand[ots]=dok[kolmik][ots][1]
  for u in uuritavad: 
   if lisand[u]:
    loetelu[u].append(lisand[u])
  
f2=open("osakaalud1.json", "w", encoding="utf-8")
json.dump(loetelu, f2)
f2.close()
for u in uuritavad:
  kogus=len(loetelu[u])
  summa=sum(loetelu[u])
  keskmine=summa/kogus
  vahim=min(loetelu[u])
  suurim=max(loetelu[u])
  print(u, kogus, keskmine, vahim, suurim)
  abi=[0]*11
  for arv in loetelu[u]:
    abi[int(10*arv/suurim)]+=1
  print(abi)