python wordcloud 模块

 

闲来无事，随便玩玩~

一段文字，用结巴分词把它分成词云，用背景图片展示出来

首先安装wordcloud，看其他博主发的是pip install wordcloud， 我这边安装不成功，在https://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud下载了个镜像（看自己python是32位的还是64位的，对应下载），然后安装成功

使用之前，先了解一下相关用法

class WordCloud(object):
    def __init__(self, font_path=None, width=400, height=200, margin=2,
             ranks_only=None, prefer_horizontal=.9, mask=None, scale=1,
             color_func=None, max_words=200, min_font_size=4,
             stopwords=None, random_state=None, background_color='black',
             max_font_size=None, font_step=1, mode="RGB",
             relative_scaling=.5, regexp=None, collocations=True,
             colormap=None, normalize_plurals=True):
          pass

font_path : string //字体路径 font_path = '宋体.ttf'
 
width : int (default=400) //输出的画布宽度，默认400像素
 
height : int (default=200) //输出的画布高度，默认200像素
 
prefer_horizontal : float (default=0.90) //词语水平方向排版出现的频率，默认横排 0.9 （所以词语垂直方向排版出现频率为 0.1 ）
 
mask : nd-array or None (default=None) //如果参数为空，则使用二维遮罩绘制词云。如果 mask 非空，设置的宽高值将被忽略，遮罩形状被 mask 取代。除全白（#FFFFFF）的部分将不会绘制，其余部分会用于绘制词云。如：bg_pic = imread('读取一张图片.png')，背景图片的画布一定要设置为白色（#FFFFFF），然后显示的形状为不是白色的其他颜色。可以用ps工具将自己要显示的形状复制到一个纯白色的画布上再保存，就ok了。
 
scale : float (default=1) //按照比例进行放大画布，如设置为1.5，则长和宽都是原来画布的1.5倍。
 
min_font_size : int (default=4) //显示的最小的字体大小
 
font_step : int (default=1) //字体步长，如果步长大于1，会加快运算但是可能导致结果出现较大的误差。
 
max_words : number (default=200) //要显示的词的最大个数
 
stopwords : set of strings or None //设置需要屏蔽的词，如果为空，则使用内置的STOPWORDS
 
background_color : color value (default=”black”) //背景颜色，如  background_color='white',背景颜色为白色。
 
max_font_size : int or None (default=None) //显示的最大的字体大小
 
mode : string (default=”RGB”) //当参数为“RGBA”并且background_color不为空时，背景为透明。
 
relative_scaling : float (default=.5) //词频和字体大小的关联性
 
color_func : callable, default=None //生成新颜色的函数，如果为空，则使用 self.color_func
 
regexp : string or None (optional) //使用正则表达式分隔输入的文本
 
collocations : bool, default=True //是否包括两个词的搭配
 
colormap : string or matplotlib colormap, default=”viridis” //给每个单词随机分配颜色，若指定color_func，则忽略该方法。

 找个图片，把词云加载上去，用可视化工具展示出来，代码如下

import wordcloud
w = wordcloud.WordCloud()
w.generate("蹦沙卡拉卡")
w.to_file("随便玩玩.png")


import jieba  # 用于分词
from wordcloud import WordCloud  # 用于制作词云
from imageio import imread  # 用于设置词云形状
import matplotlib.pyplot as plt
# Matplotlib是Python的一个绘图库，是Python中最常用的可视化工具之一，可以非常方便地创建2D图表和一些基本的3D图表。

txt = r"C:\Users\dell\Desktop\wk\ccc.txt"
with open(txt, encoding="utf-8") as file:
    contents = file.read()
    result = " ".join(jieba.lcut(contents))
    # 先用jieba对名为txt的文件分词，接着用 " ".join(参数)的方法，是参数里的每一个元素（这里是词，也可以是字符）空一格
print(result)
color_mask = imread(r"C:\Users\dell\Desktop\wk\SP\test\data_clean\随便玩玩.png")  # 词云形状是圆
wordcloud = WordCloud(
    background_color="white",
    width=1080,
    height=960,
    mask=color_mask,
    font_path=r"C:\Windows\Fonts\SIMLI.TTF",  # 若是有中文的话，这句代码必须添加，不然会出现方框，不出现汉字
    max_words=1000#控制最大字体数量
)  # 这里参数的设置见参数表

# 接下来输出词云#wordcloud是上面参数变左边的变量wordcloud，全凭你设置
wordcloud.generate(result)  # 向wordcloud对象加载文本
wordcloud.to_file("ling_yu.jpg")  # 保存图像文件
plt.imshow(wordcloud)  # 显示图像的格式
plt.show()  # 显示图像