Python做词云

网上有一些现成的词云生成工具：
HTML5、tagxedo等
但是这些工具都不太好用（有的不支持中文、有的中文词频统计得莫名其妙、有的不支持自定义形状、所有的都不能自定义颜色，尽管可能有一些主题可选）。想要深度定制，必须写代码啊。

我们使用wordcloud包

# -*- coding: utf-8 -*-
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import codecs
from wordcloud import WordCloud
import random
__author__ = 'BrownWong'

"""
模块功能：
    根据已分好词的文本生成词云
"""


def color_func(word, font_size, position, orientation, random_state=None, **kwargs):
    """
    :param word:
    :param font_size:
    :param position:
    :param orientation:
    :param random_state:
    :param kwargs:
    :return:
    Description:
        调配词云中文字的颜色
    """
    return "hsl(210, 100%%, %d%%)" % random.randint(10, 80)  # 这里以蓝色为基本色调

text = codecs.open(r'C:\Users\DELL\Desktop\text.txt', 'r', encoding='utf-8').read()  # 切分好词的纯文本(我这里并没有设置分词工具，这里我是用的文本是已经切好的，并去除了停用词的文本)
phone_mask = np.array(Image.open(r'C:\Users\DELL\Desktop\xx.jpg'))  # 词云形状的来源图片
wc = WordCloud(background_color="white", max_words=2000, mask=phone_mask,
               font_path=r'C:\Users\DELL\Desktop\chinese_font\yahei.ttf', width=800, height=400)  # 需设置中文字体
wc.generate(text)
wc.recolor(color_func=color_func, random_state=3)  # 更改原先色调
plt.imshow(wc)
plt.axis("off")
plt.show()