python - wordcloud 词不适合掩码

Question

我正在尝试从数据框中创建词云，其中有一列单词和一列代表它们各自的频率。我还下载了一张我想用作 mask 的图像，例如:cloud image

我无法让 wordcloud 适合 mask - 尽管遵循了所示示例。这是我的代码:

from PIL import Image
import pandas as pd
import numpy as np
from wordcloud import WordCloud
import matplotlib.pyplot as plt

df = pd.DataFrame({'word':['time', 'car', 'Peter\'s', 'sky', 
                           'something', 'computer', 'brain', 'plants',
                           'trees', 'Alien\'s', 'gun', 'eat'],
                   'freq':[24, 16, 16, 15,
                           14, 13, 12, 10,
                           9, 9, 5, 3]})

words = ' '.join(df['word'])

cloud_image = r"C:\Users\L\Documents\Python Scripts\cloud.png"

mask = np.array(Image.open(cloud_image))

wordcloud = WordCloud(max_words=2000,
                      width=1280, 
                      height=720,
                      mask=mask)

wordcloud.generate_from_frequencies()

fig, ax = plt.subplots(figsize=(8, 6))
ax.imshow(wordcloud, interpolation='bilinear')
ax.axis("off")
ax.imshow(mask, interpolation='bilinear')
plt.show()

Answer 1

我知道这个问题很老了，但我刚刚花了一个小时试图让一张特定的图像用作 mask 。我面临的主要问题是我想用作我的 wordcloud 蒙版的图像没有白色 (RGB 255,255,255) 背景。我最终得到的解决方案:

1. 如果您的图片有明显的背景，请使用任何在线工具从您的图片中移除背景颜色。我使用了一个名为“Background Burner”的网站，它使图像的背景变得透明。然而(正如我发现的那样)，python 的 wordcloud 将透明背景视为黑色。
2. 对于无背景图像，使用任何在线工具将背景(可能是透明的)更改为白色。我使用了一个名为 LunaPic 的网站来更改我的图像。

结果是带有白色 (RGB 255,255,255) 背景的原始图像。 wordcloud 的蒙版现在将按照您的预期处理您的图像!