PYTHON 词云

 1 #！/usr/bin/env python
 2  # -*- coding:utf-8 -*-
 3 import matplotlib.pyplot as plt
 4 from wordcloud import WordCloud,STOPWORDS,ImageColorGenerator
 5 import jieba
 6 from scipy.misc import imread
 7 from os import path
 8 
 9 def word_clould(file_path,stopwords_path,font_path,save_pic_name):
10     text_from_file_with_apath = open(file_path,encoding='utf-8').read()
11 
12     wordlist_after_jieba = jieba.cut(text_from_file_with_apath)
13     stop_words = [line.strip() for line in open(stopwords_path,encoding='utf-8')]
14     wordlist_jieba=[word for word in wordlist_after_jieba if word not in stop_words]
15     wl_space_split = " ".join(wordlist_jieba)
16     # 读取mask/color图片
17     d = path.dirname(__file__)
18     #nana_coloring = imread(path.join(d,bgpicture_path))
19 
20     my_wordcloud = WordCloud(#background_color="white",
21                              width=1200,height=800,
22                              # max_font_size=30,
23                              #max_words=5000,
24                              margin = 5,
25                              #mask = nana_coloring,
26                              random_state = 30,
27                              #stopwords = STOPWORDS,
28                              font_path=font_path).generate(wl_space_split)
29 
30     #image_colors = ImageColorGenerator(nana_coloring)
31 
32     # recolor wordcloud and show
33     #my_wordcloud.recolor(color_func=image_colors)
34     plt.imshow(my_wordcloud)
35     plt.axis("off")
36     plt.show()
37 
38     my_wordcloud.to_file(path.join(d, save_pic_name))
39 
40 
41 file_path='小米6发布会.txt'
42 stopwords_path='E:\stopwords.txt'
43 #bgpicture_path="E:\wb.jpg"
44 font_path='E:\SIMHEI.TTF'
45 save_pic_name="cloudimg1.png"
46 word_clould(file_path,stopwords_path,font_path,save_pic_name)