python wordcloud 词云

首先安装 wordcloud

1 pip install wordcloud

2 pip install jieba（中文分词）

3 pip install scipy

4 pip install pillow

5 pip install matplotlib

 

 

# -*- coding: utf-8 -*-
__author__ = 'bbs.918dxs.com'
#wordcloud生成中文词云

from wordcloud import WordCloud
import codecs
import jieba
#import jieba.analyse as analyse
from scipy.misc import imread
import os
from os import path
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont

# 绘制词云
def draw_wordcloud():
#读入一个txt文件
comment_text = open('ciyun.txt','r').read()        #ciyun.txt就是数据来源
#结巴分词，生成字符串，如果不通过分词，无法直接生成正确的中文词云
cut_text = " ".join(jieba.cut(comment_text))
d = path.dirname(__file__) # 当前文件文件夹所在目录
color_mask = imread("bg.png") # bg.png背景图片
cloud = WordCloud(
#设置字体，不指定就会出现乱码  可以控制面板--》外观--》字体查找中文字体
font_path="STXINGKA.ttf",
#font_path=path.join(d,'simsun.ttc'),
#设置背景色
background_color='white',
#词云形状
mask=color_mask,
#允许最大词汇
max_words=2000,
#最大号字体
max_font_size=40
)
word_cloud = cloud.generate(cut_text) # 产生词云
word_cloud.to_file("666.jpg") #保存图片
# 显示词云图片
plt.imshow(word_cloud)
plt.axis('off')
plt.show()

 

if __name__ == '__main__':

draw_wordcloud()