计算词频并由此制作字典

Question

我想从文本文件中提取每个词，然后计算字典中的词频。

例： 'this is the textfile, and it is used to take words and count'

d = {'this': 1, 'is': 2, 'the': 1, ...}

我没有那么远，但是我看不出如何完成它。到目前为止，我的代码：

import sys

argv = sys.argv[1]
data = open(argv)
words = data.read()
data.close()
wordfreq = {}
for i in words:
    #there should be a counter and somehow it must fill the dict.

Answer 1

如果您不想使用collections.Counter，则可以编写自己的函数：

import sys

filename = sys.argv[1]
fp = open(filename)
data = fp.read()
words = data.split()
fp.close()

unwanted_chars = ".,-_ (and so on)"
wordfreq = {}
for raw_word in words:
    word = raw_word.strip(unwanted_chars)
    if word not in wordfreq:
        wordfreq[word] = 0 
    wordfreq[word] += 1

对于更好的东西，请看正则表达式。