alex/gotohack
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Status analyzer

Browse Source
This commit is contained in:
Oleg Morozenkov
2016-02-22 13:16:36 +03:00
parent 6b0b3c9300
commit 7c39389384
1 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
status-analyzer.py
View File

@@ -1,9 +1,17 @@
import json
import sys
from collections import OrderedDict
from nltk.stem.snowball import RussianStemmer
from nltk.tokenize import RegexpTokenizer
def dictWithoutOneKey(d, key):
new_d = d.copy()
new_d.pop(key)
return new_d
# load pazans
pazansGroups = None
@@ -37,8 +45,9 @@ with open(usersFileName, "r") as file:
statusStats[filteredStatusText] = statusStatsItem
# print result
with open(sys.argv[3], "w", encoding="utf-8") as file:
for item in sorted(statusStats.items(), key=lambda item: item[1]["count-boys"] + item[1]["count-girls"], reverse=True):
file.write(item[1]["full"] + "\n")
file.write("\tboys: " + str(item[1]["count-boys"]) + "\n")
file.write("\tgirls: " + str(item[1]["count-girls"]) + "\n")
destFileName = sys.argv[3]
with open(destFileName, "w", encoding="utf8") as file:
sortKeyGetter = lambda item: item[1]["count-boys"] + item[1]["count-girls"]
sortedStatues = [item[1] for item in sorted(statusStats.items(), key=sortKeyGetter, reverse=True)]
data = OrderedDict([(item["full"], dictWithoutOneKey(item, "full")) for item in sortedStatues])
file.write(json.dumps(data, ensure_ascii=False, indent=4))