Облако слов с двумя цветами фона
Я создал это на wordcloud.com, используя одну из «тем».
Мне бы хотелось сделать это с помощью библиотеки Wordcloud Python, но пока все, что я могу добиться, — это один цвет фона (то есть весь черный, а не серый и черный). Может ли кто-нибудь дать мне подсказку, как добавить дополнительный цвет фона, используя Matlab или imshow? Вот мой код:
import numpy as np
import matplotlib.pyplot as plt
from pathlib import Path
from PIL import Image
from wordcloud import WordCloud
tmp = "some text about Dynasty TV show"
alexis_mask = np.array(Image.open('resources/alexis-poster-3.png'))
print(repr(alexis_mask)) alexis_mask[alexis_mask == 0] = 255
def color_func(word, font_size, position, orientation, random_state=None,
**kwargs):
return "hsl(0, 100%, 27%)"
wc = WordCloud(background_color = "black", mask=alexis_mask, max_words=500,contour_width=2, contour_color = "black")
wc.generate(tmp)
plt.figure(figsize=(28, 20)) plt.imshow(wc.recolor(color_func=color_func, random_state=3),interpolation = "bilinear")
plt.imshow(wc)
Я пробовал начать с черно-белого изображения и с черно-белого/серого изображения. Пока ни то, ни другое не работает. Я не думаю, что это предлагается в библиотеке Wordcloud, но могу ли я сделать это с помощью imshow() после применения wordcloud? Спасибо.
Ответы 1
Вы можете нарисовать облако слов с желаемым цветом фона (например, "grey"), а затем наложить на этот график изображение однородного цвета (например, "black"), замаскированное с помощью маски облака слов.
import numpy as np
import matplotlib.pyplot as plt
from wordcloud import WordCloud
from PIL import Image
text = "some text about Dynasty TV show"
x, y = np.ogrid[:300, :300]
mask = (((x - 150)**2 + (y - 150)**2 > 130**2) * 255).astype(int)
wc = WordCloud(background_color = "grey", repeat=True, mask=mask)
wc.generate(text)
plt.axis("off")
plt.imshow(wc)
plt.imshow(np.dstack([Image.new("RGB", (wc.width,wc.height), "black"), wc.mask]))
plt.show()
Минимальный пример подходит для случая, когда маска содержит только 0 и 255. В общем случае 1D-маски это будет (wc.mask == 255) * 255, для маски RGBA — ((wc.mask[:,:,0:3] == [255, 255, 255]) * 255)[:,:,0].
Отличный! Я вижу, как это сделали два отдельных вызова plt.imshow(), сначала обращаясь к облаку слов, а затем к маске, которая будет наложена. Потрясающий! Вот мой [результат] (kaggle.com/code/heatherpropes/wordcloud-practice) на Kaggle, если вам интересно.