Обновить u4/word_count.py

u4/word_count.py

@@ -1,8 +1,13 @@
+import string
+from collections import Counter
 def count_words(sentence):
-    cleaned = "".join(c if c.isalnum() or c == "'" else " " for c in sentence.lower())
+    text = sentence.lower()
-    counts = {}
+for mark in string.punctuation.replace("'", ""):
-    for word in cleaned.split():
+        text = text.replace(mark, " ")
+words = []
+    for word in text.split():
         word = word.strip("'")
         if word:
-            counts[word] = counts.get(word, 0) + 1
+            words.append(word)
-    return counts
+
+    return dict(Counter(words))