Python 制作简单的词云

愷龍

背景图

                               
登录/注册后可看大图

效果图

                               
登录/注册后可看大图

代码

1. from os import path
   
2. import jieba
   
3. from wordcloud import WordCloud
   
4. from PIL import Image
   
5. import numpy as np
   
6. import matplotlib.pyplot as plt
   
7. 
   
8. 
   
9. def handle_data():
   
10.     """
    
11.     处理文本数据
    
12.     :return:
    
13.     """
    
14.     # 读取数据
    
15.     with open("data.txt", "r") as f:
    
16.         txt = f.read()
    
17. 
    
18.     # 去除无效数据
    
19.     re_move = ["，", "。", " ", '\n', '\xa0']
    
20.     for i in re_move:
    
21.         txt = txt.replace(i, " ")
    
22. 
    
23.     # 使用精确分词模式
    
24.     word = jieba.lcut(txt)
    
25. 
    
26.     # 保存数据
    
27.     with open("data_handled.txt", 'w') as file:
    
28.         for i in word:
    
29.             file.write(str(i) + ' ')
    
30. 
    
31. 
    
32. def generate_image():
    
33.     """
    
34.     生成图片
    
35.     :return:
    
36.     """
    
37.     # 读取数据
    
38.     with open("data_handled.txt", "r") as file:
    
39.         txt = file.read()
    
40. 
    
41.     # 图片路径
    
42.     d = path.dirname(__file__)
    
43. 
    
44.     # 生成mask
    
45.     mask = np.array(Image.open(path.join(d, "mask.jpg")))
    
46. 
    
47.     # 生成word
    
48.     word = WordCloud(
    
49.         background_color="white",
    
50.         width=800,
    
51.         height=800,
    
52.         mask=mask,
    
53.         # 字体路径,WordCloud默认不支持中文,这里的SimHei.ttf需要下载放到系统字体库目录下
    
54.         font_path='SimHei.ttf'
    
55.     ).generate(txt)
    
56. 
    
57.     # 保存图片
    
58.     word.to_file('world_cloud.png')
    
59. 
    
60.     # 使用plt库显示图片
    
61.     plt.imshow(word)
    
62. 
    
63.     plt.axis("off")
    
64. 
    
65.     plt.show()
    
66. 
    
67. 
    
68. if __name__ == '__main__':
    
69.     handle_data()
    
70.     generate_image()
    
71. 
    

复制代码

|原文链接：https://juejin.cn/post/7008833196750012452

                               
登录/注册后可看大图

zjfyue