從文本文件中讀取並將文字頻率保存到新文本文件中，然後在新行上打印每個文件

Question

美好的一天。請幫助。使用的語言是python。下面的代碼從一個文本文件中讀取，然後將每個單詞的頻率返回到新行中。我是從這個網站https://rmtheis.wordpress.com/2012/09/26/count-word-frequency-with-python/

import re 
from collections import Counter 


def openfile(filename): 
    fh = open(filename, "r+") 
    str = fh.read() 
    fh.close() 
    return str 


def removegarbage(str): 
    # Replace one or more non-word (non-alphanumeric) chars with a space 
    str = re.sub(r'\W+', ' ', str) 
    str = str.lower() 
    return str 


def getwordbins(words): 
    cnt = Counter() 
    for word in words: 
     cnt[word] += 1 
    return cnt 


def main(filename, topwords): 
    txt = openfile(filename) 
    txt = removegarbage(txt) 
    words = txt.split(' ') 
    bins = getwordbins(words) 
    for key, value in bins.most_common(topwords): 

     print(key, value) 

main('hamlet.txt', 500) 

從上面的，它打印精美的IDE我使用（pyCharm）。但是，當我補充一點，上面的代碼下面的下面的代碼，

#Write to file 
    with open("newFile.txt", "w") as f: 
     for word in main('hamlet.txt', 500): 
      f.write(word + os.linesep) 

它打印在控制檯不錯，但顯示了一些錯誤，也是它沒寫文本文件我創建在所有。 。下面是顯示在控制檯上的示例輸出一個片段閱讀的文本文件後，它打印：

the 16 
of 12 
to 9 
search 9 
which 6 

所以，現在，我希望寫這上面的輸出到紡織的文本文件。該內容要長得多比以上。謝謝。順便說一句，誤差在控制檯上得到的是

Traceback (most recent call last): 
    File "/Users/test/PycharmProjects/Trial/trial.py", line 52, in <module> 
    for word in main("hamlet.txt", 500): 
TypeError: 'NoneType' object is not iterable 

Answer 1

，如果你想使用的功能main如圖所示，即

for word in main('hamlet.txt', 500): 

那麼函數應適應這一點。 人們可以使用例如發電機：

def main(filename, topwords): 
    txt = openfile(filename) 
    txt = removegarbage(txt) 
    words = txt.split(' ') 
    bins = getwordbins(words) 
    for key, value in bins.most_common(topwords): 
     # yield key #generate only the word, not it's frequency 
     yield key, value 

with open("newFile.txt", "w") as f: 
    for word, freq in main('hamlet.txt', 500): 
     f.write('%s\t%d\n' % (word, freq)) 

Answer 2

您需要return key, value，而不是打印出來