Create translation_hamlet.py

Author: prateekiiest

translation_hamlet.py

@@ -0,0 +1,64 @@
+# input your code here!
+
+def word_count_distribution(text):
+    
+        
+    word_counts = Counter(count_words_fast(text))
+        
+    count_distribution = dict(Counter(word_counts.values()))
+    
+    return count_distribution
+    
+    
+   
+distribution = word_count_distribution(text)
+
+
+# input your code here!
+
+def more_frequent(distribution):
+    
+    z = {}
+    for i in range(len(distribution)):
+        if(distribution[i] > i):
+            z[i] = distribution[i]
+    
+    return z
+    
+more_frequent(distribution)
+
+
+
+hamlets = pd.DataFrame(columns = ["language","distribution"])
+book_dir = "Books"
+title_num = 1
+for language in book_titles:
+    for author in book_titles[language]:
+        for title in book_titles[language][author]:
+            if title == "Hamlet":
+                inputfile = data_filepath+"Books/"+language+"/"+author+"/"+title+".txt"
+                text = read_book(inputfile)
+                distribution = word_count_distribution(text)
+                hamlets.loc[title_num] = language, distribution
+                title_num += 1
+                
+                
+                
+colors = ["crimson", "forestgreen", "blueviolet"]
+handles, hamlet_languages = [], []
+for index in range(hamlets.shape[0]):
+    language, distribution = hamlets.language[index+1], hamlets.distribution[index+1]
+    dist = more_frequent(distribution)
+    plot, = plt.loglog(sorted(list(dist.keys())),sorted(list(dist.values()),
+        reverse = True), color = colors[index], linewidth = 2)
+    handles.append(plot)
+    hamlet_languages.append(language)
+plt.title("Word Frequencies in Hamlet Translations")
+xlim    = [0, 2e3]
+xlabel  = "Frequency of Word $W$"
+ylabel  = "Fraction of Words\nWith Greater Frequency than $W$"
+plt.xlim(xlim); plt.xlabel(xlabel); plt.ylabel(ylabel)
+plt.legend(handles, hamlet_languages, loc = "upper right", numpoints = 1)
+# show your plot using `plt.show`!
+plt.show()
+