Python：制作《三国演义-三英战吕布》的词云图

划句顾

1. '''*************************************************************************
   
2.     @author:LaoGu
   
3.     @time: 2021/6/8
   
4.     @fuction:制作《三国演义-三英战吕布》的词云图
   
5. ****************************************************************************'''
   
6. import re
   
7. import jieba
   
8. import wordcloud
   
9. import collections
   
10. import numpy as np
    
11. from PIL import Image
    
12. import matplotlib.pyplot as plt
    
13. 
    
14. #读取文本
    
15. with open('三国演义-三英战吕布.txt','r') as fn:
    
16.         words=fn.read()
    
17.    
    
18. #文本预处理
    
19. pattern = re.compile(u'\t|\n|\.|-|:|;|\)|\(|\?|"')#定义正则表达式匹配模式
    
20. words = re.sub(pattern,'',words)#将符合模式的字符去除
    
21. 
    
22. #文本分词
    
23. segWords = jieba.cut(words,cut_all = False) #分词列表，精确模式分词
    
24. 
    
25. #屏蔽词汇
    
26. objWords = {} #目标词汇列表
    
27. '''
    
28. rmvWords = [u'的',u'得',u'地',u'如果']
    
29. for word in segWords:  #循环读取每个分词
    
30.     if word not in rmvWords: #如果不在屏蔽词库中
    
31.         objWords.append(word) #将分词追加到列表
    
32. '''
    
33. for word in segWords:
    
34.     if len(word) == 1:
    
35.         continue
    
36.     else:
    
37.         objWords[word] = objWords.get(word,0) + 1
    
38. items = list(objWords.items())
    
39. items.sort(key = lambda x : x[1],reverse = True)
    
40. 
    
41. 
    
42. #词频统计
    
43. word_counts = collections.Counter(objWords) #对分词做词频统计
    
44. word_counts_top20 = word_counts.most_common(20) #获取前20个最高频的词
    
45. print(word_counts_top20)
    
46. 
    
47. #词频展示
    
48. mask = np.array(Image.open('小熊.jpg'))#词频背景图
    
49. wc = wordcloud.WordCloud(
    
50.     font_path = 'C:/windows/Fonts/simhei.ttf',
    
51.     background_color = 'white',
    
52.     mask = mask,
    
53.     max_words = 30,
    
54.     max_font_size = 100
    
55.     )
    
56. wc.generate_from_frequencies(word_counts) #从字典生成词云
    
57. #image_colors = wordcloud.ImageColorGenerator(mask)#从背景图建立颜色方案
    
58. #wc.recolor(color_func = image_colors) #将词云颜色设置为背景图方案
    
59. plt.imshow(wc)#显示词云
    
60. plt.axis("off") #关闭坐标轴
    
61. plt.show()    #显示图像

复制代码

结果如下：