Skip to main content
 首页 » 编程设计

python之为文本挖掘创建词汇字典

2025年05月04日15开发

我有以下代码:

train_set = ("The sky is blue.", "The sun is bright.") 
test_set = ("The sun in the sky is bright.", 
    "We can see the shining sun, the bright sun.") 

现在我正在尝试像这样计算词频:

    from sklearn.feature_extraction.text import CountVectorizer 
    vectorizer = CountVectorizer() 

接下来我想打印词汇表。因此我这样做:

vectorizer.fit_transform(train_set) 
print vectorizer.vocabulary 

现在我没有得到输出。虽然我期待这样的事情:

{'blue': 0, 'sun': 1, 'bright': 2, 'sky': 3} 

有什么地方出了问题吗?

请您参考如下方法:

我想你可以试试这个:

print vectorizer.vocabulary_