SuperAPI AI 工具集

wordcloud-generator

设计与创作

查看项目

更新于 10-18 06:34

技能介绍

用途

本技能用于根据文本或权重数据生成词云图,核心封装了词云布局原理和wordcloud库的蒙版使用陷阱,让用户能稳定地为形状词云出图。

主要能力

  • 形状模板:内置14种形状(爱心、圆形、星形、菱形、三角形、五边形、六边形、圆环、云朵、水滴、箭头、文字蒙版、矩形、自定义图片蒙版)。
  • 配色:支持按词绑定的类别上色,也可使用colormap按词序取色;提供11种命名主题(彩虹、暖色、日落、冷色、海洋、森林、节日、灰度、商务、中国红金等)。
  • 分词:支持中英文双语,中文使用jieba分词,可加载停用词和自定义词典;英文自动小写化并过滤停用词。
  • 输入格式:JSON(词、权重、可选类别)或纯文本(自动分词计数)。
  • 输出:PNG图片,可设置背景色、字体、标题、词数、字号范围等。
  • 照片自动抠图:通过rembg将普通照片转为蒙版,一步生成词云。

使用方式

安装依赖:pip install wordcloud matplotlib numpy pillow jieba(可选抠图需pip install "rembg[cpu,cli]")。

基本命令:

python scripts/wordcloud_gen.py --input data.json --shape heart --output out.png

常用参数包括--shape、--text(文字蒙版内容)、--width/--height(矩形比例)、--colors、--colormap、--background、--theme、--title、--font、--lang、--stopwords、--userdict、--auto-cutout、--mask-mode等。

适用场景

  • 快速生成关键词/高频词可视化词云。
  • 将文字排成爱心、圆形等形状词云。
  • 中文品牌词云(推荐“品牌大字”模式:用矩形或圆形无蒙版密集云,品牌名权重设为远超其他词)。
  • 照片人物/物品轮廓词云。
  • PPT 16:9等比例矩形词云。

限制与注意事项

  • 蒙版语义:wordcloud 1.9.x中形状内部须设为0,外部设为255,否则会得到空心轮廓。
  • 中文文字蒙版效果差:CJK笔画细、字内空白多,词只能填入极小字号,推荐改用“品牌大字”模式。
  • 类别配色仅在数据带category字段时生效,否则使用colormap。
  • 非法colormap自动回退到tab10;无效输入条目会被跳过并警告。
  • 照片自动抠图首次运行需下载约170MB模型,缓存在~/.u2net。
  • 具体使用步骤和代码示例以技能文档为准,需在对应环境中运行脚本。