diff --git a/u4/word_count.py b/u4/word_count.py index c81e967..ba1bf66 100644 --- a/u4/word_count.py +++ b/u4/word_count.py @@ -1,9 +1,14 @@ def count_words(phrase): slovnik1 = {} - slova = phrase.lower().split() - for gns in slova: - cisteslovo = gns.strip(".,!:;?'\"") + zbirac = phrase.lower() + + for simbolii in r",._:;!?@#$%^&*()-+=/\|[]{}<>~`""": + zbirac = zbirac.replace(simbolii, " ") + + slova1 = zbirac.split() + for gns in slova1: + cisteslovo = gns.strip("'") if cisteslovo: slovnik1[cisteslovo] = slovnik1.get(cisteslovo, 0) + 1 - + return slovnik1