python-archieve-projects/M303/wordcloud.py

14 lines
394 B
Python

import jieba
import wordcloud
import PIL.Image as image
import numpy as np
# 这里的 xin.png 替换成自己的图片
mk = np.array(image.open('xin.png'))
text = open('红楼梦.txt', 'r', encoding='utf-8').read()
w = wordcloud.WordCloud(font_path="C:/Windows/Fonts/MSYH.TTF", background_color='white', mask = mk)
t = jieba.lcut(text)
t = ''.join(t)
w.generate(t)
w.to_file('wordcloud.png')