Python英文词云wordcloud制作

最近有对词云产生兴趣，所以对他人的代码进行了一些改良（由于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”以保存图片:"))

效果如下，如果文本内容越丰富越图像还原度越大