我用Python造了一个词云。不过这次创建的是与图像颜色匹配的词云。
使用Python wordcloud模块创建一个词云。
掩码在wordcloud模块的WordCloud类的构造函数的参数中指定。掩码以ndarray格式指定。如果为像素指定了白色，它将被遮罩。
如果要使用彩色图像作为蒙版，您可以将其颜色作为彩色映射。
ImageColorGenerator
wordcloud类之一是ImageColorGenerator。就是词云颜色生成器(基于图片颜色)。
C = wordcloud.ImageColorGenerator(image, [default_color])
| 变量 | 类型 | 内容 |
|---|---|---|
| image | 数组 | 创建词云时使用的图像。 |
| default_color | 元组 | 可选的。默认值为“None”。当画布尺寸大于为其创建颜色信息的图像时，指定颜色。（r，g，b） |
| C | 指示画布颜色信息的对象。 |
将这个对象作为WordCloud类的结构化参数来传递。
尝试一下把图像linuxmi.com.jpg做成词云。
文本自己做一个吧，我这里使用linuxmi.com.txt
为蒙板读取的图像将直接用于蒙板和颜色信息。
将ImageColorGenerator创建的对象传递给WordCloud构造函数时，指定的参数为color_func。
然后是一个执行示例。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
<span style="color: #815ba4;"><span style="background-color: #2f1e2e;">from</span></span> PIL <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">import</span></span> Image <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">import</span></span> numpy <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">as</span></span> np <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">from</span></span> wordcloud <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">import</span></span> WordCloud, ImageColorGenerator <span style="color: #8d8687;"><span style="background-color: #2f1e2e;"># 读取原始文本 </span></span> <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">with</span></span> open(<span style="color: #48b685;"><span style="background-color: #2f1e2e;">'linuxmi.txt'</span></span>, mode=<span style="color: #48b685;"><span style="background-color: #2f1e2e;">'rt'</span></span>, encoding=<span style="color: #48b685;"><span style="background-color: #2f1e2e;">'utf-8'</span></span>) <span style="color: #815ba4;"><span style="background-color: #2f1e2e;">as</span></span> fi: source_text = fi.read() <span style="color: #8d8687;"><span style="background-color: #2f1e2e;"># 创建蒙板图像 </span></span> mask_array = np.array(Image.open(<span style="color: #48b685;"><span style="background-color: #2f1e2e;">'linuxmi.com.jpg'</span></span>)) image_color = ImageColorGenerator(mask_array) <span style="color: #8d8687;"><span style="background-color: #2f1e2e;"># 创建词云 </span></span> wc = WordCloud(mask=mask_array, color_func=image_color, background_color=<span style="color: #48b685;"><span style="background-color: #2f1e2e;">'white'</span></span>).generate(source_text) <span style="color: #8d8687;"><span style="background-color: #2f1e2e;"># 直接保存在文件中的情况下，用to_file方法</span></span> wc.to_file(<span style="color: #48b685;"><span style="background-color: #2f1e2e;">'linuxmi.com.png'</span></span>) <span style="color: #8d8687;"><span style="background-color: #2f1e2e;"># 用PIL表示</span></span> <span style="color: #8d8687;"><span style="background-color: #2f1e2e;">#image_array = wc.to_array()</span></span> <span style="color: #8d8687;"><span style="background-color: #2f1e2e;">#img = Image.fromarray(image_array)</span></span> <span style="color: #8d8687;"><span style="background-color: #2f1e2e;">#img.show()</span></span> |
这是您输入的蒙版图像。
输出将如下所示。
看起来就像上图，OK，就这样。