# -*- coding: utf-8 -*- """ Created on Wed Aug 16 13:53:52 2017 @author: gzs10227 """ from os import path from scipy.misc import imread import matplotlib.pyplot as plt from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator import jieba import jieba.analyse import re from collections import Counter plt.rcParams['font.family'] = ['sans-serif'] plt.rcParams['font.sans-serif'] = ['Microsoft YaHei'] d = path.dirname(u'E:/廖庆豪/2017/201708/文本分析/') text = open(path.join(d, 'ngc.txt')).read() # read the mask / color image # taken from http://jirkavinse.deviantart.com/art/quot-Real-Life-quot-Alice-282261010 # 设置背景图片 alice_coloring = imread(path.join(d, "test.jpg")) wc = WordCloud(background_color="black", #背景颜色max_words=2000,# 词云显示的最大词数 font_path='E:/Anaconda2/Lib/site-packages/matplotlib/mpl-data/fonts/ttf/Microsoft YaHei.ttf', mask=alice_coloring,#设置背景图片 stopwords=STOPWORDS.add("said"), max_font_size=40, #字体最大值 random_state=42) text = re.sub(r'\[b\].*?\[\/b\]','',text) text = re.sub(r'\[quote\].*?\[\/pid\]','',text) text = re.sub(r'\[img\].*?\[\/img\]','',text) text = re.sub(r'\[align=center\].*?\[list\]','',text) text = re.sub(r'\[quote\].*?\[\/quote\]','',text) text = re.sub(r'\[url\].*?\[\/url\]','',text) text = re.sub(r'\[s:.*?\]','',text) text = re.sub(r'\[.*?\]','',text) text = text.replace('\t','').replace('\n','').replace(' ','') text = re.sub(r'[,:.!=?#%()#+-\/0-9]+','',text) ci = [i for i in jieba.cut(text) if len(i)>1] data = dict(Counter(ci)) ci_list = [] for k,v in data.items(): temp = (k,v) ci_list.append(temp) # 生成词云, 可以用generate输入全部文本(中文不好分词),也可以我们计算好词频后使用generate_from_frequencies函数 wc.generate(text) wc.generate_from_frequencies(data) # txt_freq例子为[('词a', 100),('词b', 90),('词c', 80)] # 从背景图片生成颜色值 image_colors = ImageColorGenerator(alice_coloring) # 以下代码显示图片 plt.imshow(wc) plt.axis("off") # 绘制词云 plt.figure() # recolor wordcloud and show # we could also give color_func=image_colors directly in the constructor plt.imshow(wc.recolor(color_func=image_colors)) plt.axis("off") # 绘制背景图片为颜色的图片 plt.figure() plt.imshow(alice_coloring, cmap=plt.cm.gray) plt.axis("off") plt.show() # 保存图片 wc.to_file(path.join(d, "tt.png"))
wordcloud词云使用
猜你喜欢
转载自blog.csdn.net/alis_xt/article/details/77249171
今日推荐
周排行