利用jieba库画词云

当前位置:

首页 > 编程开发 > 数据分析 >

利用jieba库画词云

from wordcloud import WordCloud

import matplotlib.pyplot as plt

import jieba

# 生成词云

def create_word_cloud(filename):

    with open('hongloumong.txt',encoding='utf-8') as f:

        text = f.read()

    wordlist = jieba.cut(text, cut_all=True) # 结巴分词

    wl = " ".join(wordlist)

    # 设置词云

    wc = WordCloud(

        # 设置背景颜色

        background_color="black",

        # 设置最大显示的词云数

        max_words=2000,

        # 这种字体都在电脑字体中，一般路径

        font_path='msyh.ttc',

        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('hongloumong')

二、改变图片背景：

from PIL import Image

import numpy as np

import matplotlib.pyplot as plt

import jieba

# 生成词云

def create_word_cloud(filename):

    with open('hongloumong.txt',encoding='utf-8') as f:

        text = f.read()

    wordlist = jieba.cut(text, cut_all=True) # 结巴分词

    wl = " ".join(wordlist)

    alice_mask = np.array(Image.open('xiaoxiong.jpg'))

    # 设置词云

    wc = WordCloud(

        # 设置背景颜色

        background_color="white",

        # 设置最大显示的词云数

        max_words=2000,

        # 这种字体都在电脑字体中，一般路径

        font_path='msyh.ttc',

        mask=alice_mask,

        # 设置字体最大值

        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('hongloumong')

栏目列表