【gensim--dictionary】使用方法

来源：互联网发布：远程网络教育编辑：程序博客网时间：2024/05/22 06:40

import jieba, os
import codecs
from gensim import corpora, models, similarities
from pprint import pprint
from collections import defaultdict
import sys
import pickle
reload(sys)
sys.setdefaultencoding('utf-8')
def print_dict(dict):
for key in dict:
print type(key),key,str(dict[key]),
print
def test3():
'''''''
gensim学习之Dictionary
'''
a = [['一','一','二'],['一','二','三']]
b = ['一','一','三','四','四']
dictionary = corpora.Dictionary(a)
print "########dictionary信息##########"
print str(dictionary) #
print "字典，{单词id，在多少文档中出现}"
print dictionary.dfs #字典，{单词id，在多少文档中出现}
print "文档数目"
print dictionary.num_docs #文档数目
print "dictionary.items()"
print_dict(dict(dictionary.items())) #
print "字典，{单词id，对应的词}"
print_dict(dictionary.id2token) #字典，{单词id，对应的词}
print "字典，{词，对应的单词id}"
print_dict(dictionary.token2id) #字典，{词，对应的单词id}
print "所有词的个数"
print dictionary.num_pos #所有词的个数
print "每个文件中不重复词个数的和"
print dictionary.num_nnz #每个文件中不重复词个数的和
print "########doc2bow##########"
#dictionary.add_documents([b])
#allow_update->更新当前字典；return_missing->返回字典中不存在的词
#result为b文章转换得到的词袋，列表[(单词id，词频)]
result, missing = dictionary.doc2bow(b, allow_update=False, return_missing=True)
print "词袋b，列表[(单词id，词频)]"
print result
print "不在字典中的词及其词频，字典[(单词，词频)]"
print_dict(missing)
print "########bow信息##########"
for id, freq in result:
print id, dictionary.id2token[id], freq
print "########dictionary信息##########"
#过滤文档频率大于no_below，小于no_above*num_docs的词
dictionary.filter_extremes(no_below=1, no_above=0.5, keep_n=10)
return
test3()

阅读全文

0 0