用 python 开发一个词云生成库
时间: 2023-08-04 16:11:04 浏览: 91
好的,以下是一个简单的词云生成库的示例代码:
```python
import numpy as np
from PIL import Image
from wordcloud import WordCloud, ImageColorGenerator
def generate_word_cloud(text, mask_path, font_path, output_path):
# 加载遮罩图像
mask = np.array(Image.open(mask_path))
# 创建 WordCloud 对象
wc = WordCloud(background_color="white", mask=mask, font_path=font_path,
max_words=2000, max_font_size=100, random_state=42)
# 生成词云图像
wc.generate(text)
# 从图像中生成颜色
image_colors = ImageColorGenerator(mask)
# 将词云图像与颜色结合
wc.recolor(color_func=image_colors)
# 保存生成的词云图像
wc.to_file(output_path)
```
该函数接受四个参数:
- `text`:要生成词云的文本字符串。
- `mask_path`:用于遮罩词云的图像文件路径。
- `font_path`:用于渲染词云的字体文件路径。
- `output_path`:生成的词云图像文件路径。
示例用法:
```python
generate_word_cloud("hello world, hello python", "mask.png", "font.ttf", "output.png")
```
其中,`mask.png` 是遮罩图像文件,`font.ttf` 是字体文件,`output.png` 是生成的词云图像文件。
阅读全文