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

读写文件

把待读取的文本存在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')    # 保存到本地图片文件