stef_buwalda/whatsapp-wordcloud
1
0
Fork 0
mirror of https://github.com/StefBuwalda/whatsapp-wordcloud.git synced 2025-11-01 12:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added chat% as print to terminal

Browse Source
This commit is contained in:
Stef
2025-04-18 14:28:17 +02:00
parent 2c82589a10
commit 9cf6f679b3
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
run.py
View File

@@ -58,12 +58,20 @@ wordcloud = WordCloud(
makedirs("output", exist_ok=True) makedirs("output", exist_ok=True)
worddict: dict[str, int] = {}
total = 0
for author in author_words.keys(): for author in author_words.keys():
words = author_words.get(author) words = author_words.get(author)
if words: if words:
worddict[author] = len(words)
total += len(words)
for word in words: for word in words:
word_count_dicts[word] = word_count_dicts.get(word, 0) + 1 word_count_dicts[word] = word_count_dicts.get(word, 0) + 1
test = wordcloud.generate_from_frequencies( # type: ignore test = wordcloud.generate_from_frequencies( # type: ignore
word_count_dicts word_count_dicts
) )
test.to_file("output/" + author + ".png") # type: ignore test.to_file("output/" + author + ".png") # type: ignore
for author in worddict.keys():
count = worddict[author]
print(f"{author}: {count}/{total} ({round(count/total*100, ndigits=1)}%)")