`WordCloud`如何设置背景图?
时间: 2024-09-15 16:11:30 浏览: 50
在`WordCloud`中设置背景图是为了增加词云的独特性和视觉吸引力。你可以通过传递一个图像数组或者路径给`mask`参数来实现。这里有一个简单的例子:
```python
import numpy as np
from PIL import Image
from wordcloud import WordCloud
# 加载中国地图图片
background_image = np.array(Image.open('china.jpg'))
# 创建词云对象
wc = WordCloud(
font_path='simhei.ttf', # 字体文件路径
background_color='white', # 背景色,默认白色,如果你想保持图片原有色调可以用'random'
mask=background_image, # 设置背景图片
width=background_image.shape[1], # 图片宽度
height=background_image.shape[0], # 图片高度
)
# 生成词云
wc.generate_from_text('...')
# 然后显示词云
wc.to_file('my_china_wordcloud.png') # 保存为图片
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.show() # 显示在matplotlib图形窗口中
```
在这个示例中,`mask`参数接受的`background_image`是一个numpy数组,它应该已经被调整成适合词云大小的灰度图像。如果你传递的是图片文件路径,`np.array(Image.open())`会自动加载并转换图片。
阅读全文