《三国演义》——词云制作

 1 # -*- coding: utf-8 -*-
 2 """
 3 Created on Mon Apr  6 22:45:36 2020
 4 
 5 @author: 49594
 6 """
 7 
 8 # coding:utf-8
 9  
10  
11 from wordcloud import WordCloud
12 import matplotlib.pyplot as plt
13 import jieba
14 
15 
16 
17  
18 # 生成词云
19 def create_word_cloud(filename):
20     
21     text = open("{}.txt".format(filename), 'rb' ).read()
22 #结巴分词
23     
24     
25     wordlist = jieba.cut(text, cut_all=True) # 结巴分词
26     wl = " ".join(wordlist)
27    
28    
29     # 设置词云
30     wc = WordCloud(
31         # 设置背景颜色
32         background_color="black",
33         # 设置最大显示的词云数
34       
35         max_words=2000,
36         # 这种字体都在电脑字体中，一般路径
37         font_path='simsun.ttf',
38         height=1200,
39         width=1600,
40         # 设置字体最大值
41         max_font_size=100,
42         # 设置有多少种随机生成状态，即有多少种配色方案
43         random_state=100,
44     )
45  
46     myword = wc.generate(wl)  # 生成词云
47     # 展示词云图
48     plt.imshow(myword)
49     plt.axis("off")
50     plt.show()
51     wc.to_file('py_book.png')  # 把词云保存下
52  
53  
54 if __name__ == '__main__':
55     create_word_cloud('三国演义')
56 

 1 # -*- coding: utf-8 -*-
 2 """
 3 Created on Mon Apr  6 22:45:36 2020
 4 
 5 @author: 49594
 6 """
 7 
 8 # coding:utf-8
 9  
10  
11 from wordcloud import WordCloud
12 import matplotlib.pyplot as plt
13 import jieba
14 import numpy as np
15 from PIL import Image
16 
17 
18  
19 # 生成词云
20 def create_word_cloud(filename):
21     
22     text = open("{}.txt".format(filename), 'rb' ).read()
23 #结巴分词
24     
25     
26     wordlist = jieba.cut(text, cut_all=True) # 结巴分词
27     wl = " ".join(wordlist)
28     cloud_mask = np.array(Image.open(".png"))
29 
30    
31     # 设置词云
32     wc = WordCloud(
33         # 设置背景颜色
34         background_color="black",
35         # 设置最大显示的词云数
36         mask=cloud_mask,
37         #设置背景图片
38         max_words=2000,
39         # 这种字体都在电脑字体中，一般路径
40         font_path='simsun.ttf',
41         height=1200,
42         width=1600,
43         # 设置字体最大值
44         max_font_size=100,
45         # 设置有多少种随机生成状态，即有多少种配色方案
46         random_state=100,
47     )
48  
49     myword = wc.generate(wl)  # 生成词云
50     # 展示词云图
51     plt.imshow(myword)
52     plt.axis("off")
53     plt.show()
54     wc.to_file('py_book.png')  # 把词云保存下
55  
56  
57 if __name__ == '__main__':
58     create_word_cloud('三国演义')
59