美文网首页
Python jieba 去除停用词 生成词云图

Python jieba 去除停用词 生成词云图

作者: zestloveheart | 来源:发表于2018-10-11 19:24 被阅读0次

读写文件

把待读取的文本存在info.txt中,content类型为str

with open('info.txt', 'r', encoding="UTF-8") as file1:  # with as操作读取文件很ok
    content = "".join(file1.readlines())

待写入文件为 output.txt,content_after为待写入字符串

with open('output.txt', 'w', encoding='utf-8') as file2:
    file2.write(content_after+"\n")

分词

# 调用jieba.cut
sentence_seged = jieba.cut(content)

去除停用词

  • 建立停用词表
    将停用词表放在stop.txt中,一行一个词
    # stopwords为停用词list
    stopwords = [line.strip() for line in open('stop.txt', 'r', encoding='utf-8').readlines()]
    
  • 遍历去除停用词
    outstr = '' # 待返回字符串
    
     for word in sentence_seged:
        if word not in stopwords:
            outstr += word + " "
    

生成词云图

images = Image.open("something.png") # 打开保存的图片
maskImages = np.array(images) # 并用numpy转换
wc = WordCloud(font_path="msyh.ttc", background_color="white", max_words=100, max_font_size=100).generate(content_after) # 生成词云图
wc.to_file('wordCloudPic.png')    # 保存到本地图片文件

相关文章

网友评论

      本文标题:Python jieba 去除停用词 生成词云图

      本文链接:https://www.haomeiwen.com/subject/ouhiaftx.html