diff --git a/strings/word_occurrence.py b/strings/word_occurrence.py index 5a18ebf771e4..420aeba74df7 100644 --- a/strings/word_occurrence.py +++ b/strings/word_occurrence.py @@ -1,23 +1,22 @@ # Created by sarathkaul on 17/11/19 # Modified by Arkadip Bhattacharya(@darkmatter18) on 20/04/2020 -from collections import defaultdict -def word_occurrence(sentence: str) -> dict: +def word_occurrence(sentence: str) -> dict[str, int]: """ >>> from collections import Counter >>> SENTENCE = "a b A b c b d b d e f e g e h e i e j e 0" - >>> occurence_dict = word_occurrence(SENTENCE) - >>> all(occurence_dict[word] == count for word, count + >>> occurrence_dict = word_occurrence(SENTENCE) + >>> all(occurrence_dict[word] == count for word, count ... in Counter(SENTENCE.split()).items()) True >>> dict(word_occurrence("Two spaces")) {'Two': 1, 'spaces': 1} """ - occurrence: defaultdict[str, int] = defaultdict(int) - # Creating a dictionary containing count of each word + occurrence: dict[str, int] = {} + for word in sentence.split(): - occurrence[word] += 1 + occurrence[word] = occurrence.get(word, 0) + 1 return occurrence