Python生成词云图

1、安装Python包

pip install jieba
pip install numpy
pip install matplotlib
pip wordcloud

2、安装wordcloud失败解决方法,手动安装

下载安装包,链接
pip install 安装包本地位置

3、主程序,bg.png和test.txt和主程序在同一目录,bg.png为生成词云的样子,test.txt为词云的文字
import jieba  # 分词
from wordcloud import WordCloud  # 词云图相关
import matplotlib.pyplot as plt
import matplotlib.colors as colors  # 处理图片相关内容
import numpy
from PIL import Image


# 生成词云方法
def get_wcloud():
    # 读取小说内容
    with open(r'test.txt', 'r', encoding='UTF-8') as txt:
        data = txt.read()
    # 将文件中所有文字分词
    words_list = jieba.lcut(data)
    # 用空格分隔词语
    tokenstr = ' '.join(words_list)
    # 打开背景图片
    color_mask = numpy.array(Image.open('bag.png'))
    # 自定义文字颜色
    colormaps = colors.ListedColormap(['#0D8EEC'])
    # 生成词云(默认样式)
    # mywc1 = WordCloud().generate(tokenstr)
    # 生成词云(自定义样式)
    mywc1 = WordCloud(
        scale=4,
        font_path="C:/Windows/Fonts/simfang.ttf",
        mask=color_mask,  # 指定背景图形状
        colormap=colormaps,  # 指定颜色
        background_color='white',  # 指定背景颜色,默认黑色
        width=1920,  # 指定宽度
        height=1080  # 指定高度
    ).generate(tokenstr)
    # 显示词云
    plt.imshow(mywc1)
    plt.axis('off')
    plt.show()
    mywc1.to_file('success.png')  # 生成词云图片


if __name__ == '__main__':
    get_wcloud()
4、另外一种词云生成方法,在线词云生成工具,链接
相关推荐
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页