Python做词云

来源:互联网 发布:逆袭网络剧可以在哪看 编辑:程序博客网 时间:2024/05/17 23:04

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

我们使用wordcloud

# -*- coding: utf-8 -*-from PIL import Imageimport numpy as npimport matplotlib.pyplot as pltimport codecsfrom wordcloud import WordCloudimport 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()
1 0