Python的第三方库—WordCloud

Posted 2021-09-21 Harris-H

Python的第三方库—WordCloud

0.常用参数

1.快速开始

# pip install pyecharts -i https://pypi.douban.com/simple  安装词云库
 
import wordcloud
c = wordcloud.WordCloud()         # 步骤一：配置对象参数
c.generate("worldcloud python python python python")   # 步骤二：加载词云文本
c.to_file("1.png")          # 步骤三：输出词云文件 默认大小为高200像素，宽400像素

样例2

import jieba  # 中文分词插件
import wordcloud

if __name__ == '__main__':
    file = open("三国演义.txt", "r", encoding="utf-8")
    t = file.read()
    file.close()

    ls = jieba.lcut(t)  # 找到所有词
    txt = " ".join(ls)  # 空格分词
    w = wordcloud.WordCloud(width=700, height=500,  # 词云图片的宽度和高度设置
                            background_color="white",  # 背景颜色
                            font_path='msyh.ttc',  # 字体
                            stopwords='',  # 要过滤的词的集合
                            mode='RGB',  # 默认为RGB，若设置为RGBA且background_color="None",则会产生透明背景的词云图片
                            max_font_size=90)  # 最大字号
    w.generate(txt)
    w.to_file("picture/test-02.png")

2.自定义形状

通过设置$mask$参数即可。

注意$imread$不太好用。

这里使用PIL的Image ，然后利用$numpy$转换为$array$。

import numpy as np
from PIL import Image
mask = np.array(Image.open("background/heart.png"))


# 完整版
import jieba  # 中文分词插件
import numpy as np
import wordcloud
from PIL import Image
if __name__ == '__main__':
    file = open("三国演义.txt", "r", encoding="utf-8")
    t = file.read()
    file.close()
    mask = np.array(Image.open("background/heart.png"))
    ls = jieba.lcut(t)  # 找到所有词
    txt = " ".join(ls)  # 空格分词
    w = wordcloud.WordCloud(width=700, height=500,  # 词云图片的宽度和高度设置
                            background_color="white",  # 背景颜色
                            font_path='msyh.ttc',  # 字体
                            mask=mask,
                            stopwords='',  # 要过滤的词的集合
                            mode='RGB',  # 默认为RGB，若设置为RGBA且background_color="None",则会产生透明背景的词云图片
                            max_font_size=90)  # 最大字号
    w.generate(txt)
    w.to_file("picture/test-02-heart.png")

3.参考文章

传送门１

传送门２

WordCloud-github

WordCloud官网