最近有对词云产生兴趣,所以对他人的代码进行了一些改良(由于BUG和使用方式不太合意)从而达到了词云制作的目的
#导入wordcloud模块和matplotlib模块
import wordcloud as WD
import matplotlib.pyplot as plt
from scipy.misc import imread
#读取一个txt文件
text = open(input('请输入“文本名.txt:”'),'r').read()
#读入背景图片
bg_pic = imread(input("请输入“图片名.XXX :”"))
#生成词云
wordcloud = WD.WordCloud(mask=bg_pic,background_color='white',scale=1.5).generate(text)
#显示词云图片
plt.imshow(wordcloud)
plt.axis('off')
plt.show()
#保存图片
wordcloud.to_file(input("请输入保存“取名.jpg”以保存图片:"))
效果如下,如果文本内容越丰富越图像还原度越大