用途
本技能用于根据文本或权重数据生成词云图,核心封装了词云布局原理和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。
- 具体使用步骤和代码示例以技能文档为准,需在对应环境中运行脚本。