使用tf-idf（Gensim）获取语料库中最重要的词

Question

我正在计算tf-idf如下。

texts=['human interface computer',
 'survey user computer system response time',
 'eps user interface system',
 'system human system eps',
 'user response time']

dictionary = corpora.Dictionary(texts)
corpus = [dictionary.doc2bow(text) for text in texts]
tfidf = models.TfidfModel(corpus)
corpus_tfidf = tfidf[corpus]
analyzedDocument = namedtuple('AnalyzedDocument', 'word tfidf_score')
d=[]
for doc in corpus_tfidf:
    for id, value in doc:
        word = dictionary.get(id)
        score = value
        d.append(analyzedDocument(word, score))

但是，现在我想使用idf值最高的单词来确定语料库中最重要的3个单词。 请让我知道该怎么做？

Answer 1

假设您的清单确定正确，则应该能够按照以下方式进行排列：最上方：

from operator import itemgetter

然后在底部：

e=sorted(d, key=itemgetter(1))
top3 = e[:3]
print(top3)