ozbolt/luscenje_struktur
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added limit to minimum frequency = 10 + Ordered by frequency

Browse Source
This commit is contained in:
Luka 2020-08-21 15:05:30 +02:00
parent edea80e6e0
commit e38ff4c7b0
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/recalculate_statistics.py
View File

@ -174,6 +174,12 @@ def main(args):
write_file_path = os.path.join(args.output, file_name) write_file_path = os.path.join(args.output, file_name)
with open(read_file_path, 'r') as rf, open(write_file_path, 'w') as wf: with open(read_file_path, 'r') as rf, open(write_file_path, 'w') as wf:
original_text, stats = get_new_stats(rf) original_text, stats = get_new_stats(rf)
freq_pos = original_text[0].index('Frequency')
original_text = [original_text[0]] + [l for l in original_text[1:] if int(l[freq_pos]) >= 10]
if len(original_text) > 1:
original_text = [original_text[0]] + sorted(original_text[1:], key=lambda x: -1 * int(x[freq_pos]))
else:
original_text = [original_text[0]]
write_new_stats(wf, original_text, stats, file_name, word_order) write_new_stats(wf, original_text, stats, file_name, word_order)
if __name__ == '__main__': if __name__ == '__main__':