from os import path;from PIL import Image;import numpy as np;import matplotlib.pyplot as plt;from wordcloud import WordCloud, STOPWORDS
d = path.dirname(__file__)
# 获取组成词云的文本alice.txt
text = open(path.join(d, 'alice.txt'),encoding='utf-8').read()
#读取图片,可以把想要输出的图片设置为alice_mask.png
alice_mask = np.array(Image.open(path.join(d, "alice_mask.png")))
stopwords = set(STOPWORDS)
stopwords.add("said")
#设置词云的一些属性,font_path设置为中文支持,否则读取中文会乱码
wc = WordCloud(background_color="white", max_words=1000000000, mask=alice_mask,max_font_size=100,width=1800,height=900,min_font_size=1,
stopwords=stopwords,font_path='C:\Windows\Fonts\SIMYOU.TTF')
# 生成词云
wc.generate(text)
#保存到本地
wc.to_file(path.join(d, "alice.png"))
#展示
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.figure()
plt.imshow(alice_mask, cmap=plt.cm.gray, interpolation='bilinear')
plt.axis("off")
plt.show()
特定图形词云
最新推荐文章于 2024-04-20 15:26:30 发布