python 简单实现jieba库分词制作词云

落日映苍穹つ 2022-03-21 08:18 251阅读 0赞

## 准备工作： ##

python版本：python 2

ide：pycharm2018

pip 安装 jieba ， wordcloud ，matplotlib

\#保存至脚本同一级位置

百度一段词典：红楼梦，水浒传

下载中文转换的ttf文件：[https://s3-us-west-2.amazonaws.com/notion-static/b869cb0c7f4e4c909a069eaebbd2b7ad/simsun.ttf][https_s3-us-west-2.amazonaws.com_notion-static_b869cb0c7f4e4c909a069eaebbd2b7ad_simsun.ttf]

## 实现： ##

# coding:utf-8
    
    
    from wordcloud import WordCloud
    import matplotlib.pyplot as plt
    import jieba
    
    
    # 生成词云
    def create_word_cloud(filename):
        text = open("{}.txt".format(filename)).read()
        
        wordlist = jieba.cut(text, cut_all=True) # 结巴分词
        wl = " ".join(wordlist)
    
        # 设置词云
        wc = WordCloud(
            # 设置背景颜色
            background_color="black",
            # 设置最大显示的词云数
            max_words=2000,
            # 这种字体都在电脑字体中，一般路径
            font_path='simsun.ttf',
            height=1200,
            width=1600,
            # 设置字体最大值
            max_font_size=100,
            # 设置有多少种随机生成状态，即有多少种配色方案
            random_state=100,
        )
    
        myword = wc.generate(wl)  # 生成词云
        # 展示词云图
        plt.imshow(myword)
        plt.axis("off")
        plt.show()
        wc.to_file('img_book.png')  # 把词云保存下
    
    
    if __name__ == '__main__':
        create_word_cloud('mytext')

## 展示： ##

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDkwMjUyNw_size_16_color_FFFFFF_t_70][]

加载背景图片生成词云

from wordcloud import WordCloud
    import matplotlib.pyplot as plt
    import jieba
    import numpy as np
    from PIL import Image
    
    
    # 生成词云
    def create_word_cloud(filename):
        text = open("{}.txt".format(filename)).read()
        # 结巴分词
        wordlist = jieba.cut(text, cut_all=True)
        wl = " ".join(wordlist)
    
        # 加载背景图片
        cloud_mask = np.array(Image.open("bg.png"))
    
        # 设置词云
        wc = WordCloud(
            # 设置背景颜色
            background_color="black",
            #背景图片
            mask=cloud_mask,
            # 设置最大显示的词云数
            max_words=2000,
            # 这种字体都在电脑字体中，一般路径
            font_path='simsun.ttf',
            height=1200,
            width=1600,
            # 设置字体最大值
            max_font_size=100,
            # 设置有多少种随机生成状态，即有多少种配色方案
            random_state=100,
        )
    
        myword = wc.generate(wl)  # 生成词云
        # 展示词云图
        plt.imshow(myword)
        plt.axis("off")
        plt.show()
        wc.to_file('py_book.png')  # 把词云保存下
    
    
    if __name__ == '__main__':
        create_word_cloud('mytext')

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDkwMjUyNw_size_16_color_FFFFFF_t_70 1][]

[https_s3-us-west-2.amazonaws.com_notion-static_b869cb0c7f4e4c909a069eaebbd2b7ad_simsun.ttf]: https://s3-us-west-2.amazonaws.com/notion-static/b869cb0c7f4e4c909a069eaebbd2b7ad/simsun.ttf
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDkwMjUyNw_size_16_color_FFFFFF_t_70]: /images/20220321/6b14f0ec276340138526b3a48c951cfe.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MDkwMjUyNw_size_16_color_FFFFFF_t_70 1]: /images/20220321/bad251883a644bb0b40fe85b961559df.png