28 lines
809 B
Python
28 lines
809 B
Python
from collections import defaultdict
|
|
|
|
def count_words(sentance: str) -> dict:
|
|
sent_list = list(sentance)
|
|
words = defaultdict(int)
|
|
|
|
delims = [":", "?", "!", "\n", "\t", ",", "."]
|
|
del_chars = ['"']
|
|
|
|
forbidden = delims + del_chars + [" ", "'", ""]
|
|
|
|
for i, ch in enumerate(sent_list):
|
|
if(ch in delims):
|
|
sent_list[i] = " "
|
|
elif(ch in del_chars):
|
|
sent_list[i] = ""
|
|
if(ch == "'"):
|
|
if((sent_list[max(i-1, 0)] not in forbidden) and (sent_list[min(i+1, len(sent_list)-1)] not in forbidden)):
|
|
sent_list[i] = '^'
|
|
else:
|
|
sent_list[i] = ""
|
|
|
|
for w in ''.join(sent_list).lower().split(" "):
|
|
if w == "":
|
|
continue
|
|
words[w.replace("^", "'")] += 1
|
|
|
|
return dict(words) |