python如何画云_用python绘制词云

import sys

from wordcloud import WordCloud

import jieba

from scipy.misc import imread

import matplotlib.pyplot as plt

reload(sys)

sys.setdefaultencoding("utf-8")

# 读入文本

comment_text = open('test.txt','r').read()

# 结巴分词，生成字符串，如果不通过分词，无法直接生成正确的中文词云

cut_text = " ".join(jieba.cut(comment_text))

print cut_text

# 读取背景图片

color_mask = imread('background.png')

cloud = WordCloud(

# 设置背景色

background_color='black',

# 词云形状

mask=color_mask,

#允许最大词汇

max_words=2000,

# 设置字体，不设置可能出现乱码

font_path="STFANGSO.ttf",

#最大号字体

max_font_size=100,

)

# 产生词云

word_cloud = cloud.generate(cut_text)

# 保存图片

word_cloud.to_file('after.jpg')

# 显示词云图片

plt.imshow(word_cloud)

plt.show()

效果如下：