#-*-coding:utf-8 -*-

from os import path
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
#用来画词云的。
import matplotlib.pyplot as plt
#用来解析特殊的图片模板，numpy也可以的。
from scipy.misc import imread

#第一步读你要制作成词云的源文件，少年记得用‘r’去读，手抖用了次‘w+’结果文件让我清空了。。。
dir=r‘C:\Users\cchen\Desktop\MPLS_2.txt‘
text=open(dir,‘r‘).read()
#用导入的scipy的imread去解析模板文件,图片越大解析越慢
alice_coloring = imread(r‘C:\Users\cchen\Desktop\cloud.jpg‘)
#print alice_coloring 所料不错，是个挺大的矩阵。大到卡住了

#生成词云，这里可以用到很多的参数
#background_color顾名思义，是背景的颜色
#max_words 词云显示的最大词数
#mask 主角，我们的个人定制的模板
#max_font_size 字体的最大值
#stopwords 去除一些常用词汇，比如a,is,the之类的，你可以用add加上你不想显示的词。
#font_path=‘./font/cabin-sketch.bold.ttf设置字体
wc = WordCloud(background_color="white", max_words=600, mask=alice_coloring,max_font_size=200, random_state=42,stopwords=STOPWORDS.add("said"))
wc.generate(text)

#show出来定制化的词云,颜色是随机的五颜六色，第一个图就是啦
plt.figure()
plt.imshow(wc)
plt.axis("off")



# 我这么低调的人肯定要用原图的黑色啊
# 可以用color_func=image_colors来重新画
# 根据模板获取颜色
image_colors = ImageColorGenerator(alice_coloring)
#获取到的颜色模板，第二个图
plt.imshow(wc.recolor(color_func=image_colors))
plt.axis("off")
#低调的词云，第三个图
plt.figure()
plt.imshow(alice_coloring, cmap=plt.cm.gray)
plt.axis("off")
plt.show()