弄好英文词云之后,我突然有了将程序一点直接生成词云的想法,于是做出了以下代码:
#导入wordcloud模块和matplotlib模块
import wordcloud as WD
import matplotlib.pyplot as plt
from scipy.misc import imread
import glob,os
#读取一个txt文件
for wdfile in glob.glob('*.txt'):
text = open(wdfile,'r').read()
#读入背景图片
for pcfile in glob.glob('*.jpg'):
file,ext = os.path.splitext(pcfile)
bg_pic = imread(pcfile)
#生成词云
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(file+"词云版.jpg")
这样一来,只需双击程序,便可以将同文件夹中图片与文本结合生成词云了。