用WordCloud绘制词云

文章目录

初步认识

wordcloud是词云绘图模块,封装了WordCloud词云类,是词云的基本载体。在新建一个词云之后,通过generate装载用以生成词云的字符串,最后用to_file把词云图保存到文件中,例如

python 复制代码
from wordcloud import WordCloud

w = WordCloud()
txt = "tinycool cool tiny Python WordCloud word cloud word not world"
w.generate(txt)
w.to_file("pywordcloud.png")

效果如下

基本参数

仅就这张图像来说,我们能感受到的特性大概有

  • 词云图的宽度、高度
  • 背景颜色,文字颜色
  • 字体
  • 单词长度,单词个数

这些特性可受到下列参数调控

参数 说明 备注
font_path 字体文件路径 otf或ttf文件
width 词云宽度 默认400
height 词云高度 默认200
min_font_size 最小文字尺寸 默认4
max_font_size 最大文字尺寸 默认为图像高度
relative_scaling 词频对尺寸的影响 默认"auto"
font_step 字体步长 默认1
max_words 最大单词数 默认200
min_word_length 最短单词长度 默认0
background_color 背景色 默认"black"
mode 颜色格式 默认RGB
colormap 颜色映射
repeat 是否重复单词 默认False

下面对这些参数稍作更改,并

python 复制代码
# txt即本文前面所有文字,因为太多就不写了
txt = r'''
'''
dct = dict(width=600, height=300,
    font_path = r"C:\Windows\Fonts\simhei.ttf",
    min_font_size = 10,
    max_font_size = 100,
    max_words = 20,
    min_word_length = 2,
    background_color = "white",
    colormap = "jet")

w = WordCloud(**dct)
w.generate(txt)
w.to_file("pywordcloud.png")

效果如下,由于最多只显示20个单词,而且尺寸最小是10,最大100,所以不太好看。

掩模参数

所谓掩模,就是词云组成的图案,主要由以下参数调控。

参数 说明 备注
mask 掩模数组 默认None
contour_width 掩模宽度 默认为0
contour_color 掩模颜色 默认"black"

掩模数组由0和1构成,词云只在掩模数组为1的位置显示。下面画一个椭圆作为掩模数组,椭圆圆周的方程为

( x − 300 ) 2 28 0 2 + ( y − 150 ) 2 14 0 2 = 1 \frac{(x-300)^2}{280^2}+\frac{(y-150)^2}{140^2}=1 2802(x−300)2+1402(y−150)2=1

x = 300 ± 28 0 2 − ( 2 y − 300 ) 2 x = 300\pm\sqrt{280^2-(2y-300)^2} x=300±2802−(2y−300)2

python 复制代码
import numpy as np
y, x = np.indices([300,600])
delta = np.sqrt(280**2-(2*y-300)**2)
mask = (x<300+delta) & (x>300-delta)
mask = (1-mask)*254+1

w = WordCloud( mask = mask, contour_width=100,
    font_path = r"C:\Windows\Fonts\simhei.ttf")
w.generate(txt)
w.to_file("pywordcloud.png")

效果如下

相关推荐
Deng94520131427 分钟前
基于Python的旅游数据可视化应用
python·numpy·pandas·旅游·数据可视化技术
2401_8786247930 分钟前
pytorch 自动微分
人工智能·pytorch·python·机器学习
胖达不服输33 分钟前
「日拱一码」021 机器学习——特征工程
人工智能·python·机器学习·特征工程
screenCui2 小时前
macOS运行python程序遇libiomp5.dylib库冲突错误解决方案
开发语言·python·macos
小眼睛羊羊2 小时前
pyinstaller打包paddleocr
python
java1234_小锋2 小时前
基于Python的旅游推荐协同过滤算法系统(去哪儿网数据分析及可视化(Django+echarts))
python·数据分析·旅游
蓝婷儿2 小时前
Python 机器学习核心入门与实战进阶 Day 4 - 支持向量机(SVM)原理与分类实战
python·机器学习·支持向量机
%d%d22 小时前
python 在运行时没有加载修改后的版本
java·服务器·python
amazinging3 小时前
北京-4年功能测试2年空窗-报培训班学测开-第四十七天
python·学习·selenium
Freak嵌入式3 小时前
一文速通 Python 并行计算:13 Python 异步编程-基本概念与事件循环和回调机制
开发语言·python·嵌入式·协程·硬件·异步编程