第一次用Python制作词云

用了一天时间终于做好第一张算是满意的词云了。网上的教程看了不少，还是有不少坑的。
下面贴上我用的代码：

__author__ = 'Solarzhou'
#!/usr/bin/env python
#-*- coding: utf-8 -*-
from os import path
from wordcloud import WordCloud, STOPWORDS
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
import random
import os
def grey_color_func(word, font_size, position, orientation, random_stata=None, **kwargs):
    return 'hsl(0, 0%%, %d%%)' % random.randint(60, 100)
    #当前文件夹所在目录
d = path.dirname(__file__)
#读取背景图片
mask = np.array(Image.open(path.join(d, 'picture\LBJ.jpg')))
#读取一个txt文件
text = open(u'File\honor2.txt').read()
#停用词词库
stopwords = set(STOPWORDS)
stopwords.add("int")
stopwords.add("ext")
#设置词云属性，其中第一个参数非常重要，若没有正确设置，
#将不能解析中文，出现乱码
wc = WordCloud(font_path='./fonts/simhei.ttf', max_words=2000, mask=mask,
               stopwords=stopwords, margin=10,
               random_state=1).generate(text)

default_colors = wc.to_array()
plt.title("Custom colors")
plt.imshow(wc.recolor(color_func=grey_color_func, random_state=3))
wc.to_file('001.png')
plt.axis("off")
#显示生成的ciyun
plt.figure()
plt.imshow(default_colors)
plt.axis("off")
plt.show()

代码不多，也容易懂，不过在这之前需要安装wordcloud，numpy，pil和matplotlib包。
读取中文文档还需要额外处理。
运行之后生成效果图：

原图：
上述代码同样适用于读取中文文档，都是亲测过的，这里就不贴图了。更详细地步骤我在微信里也发过，有问题的可以戳开。原文链接 我会及时解答的。