头歌平台python参考答案词云绘制
时间: 2023-06-29 11:19:10 浏览: 263
好的,关于头歌平台Python参考答案中的词云绘制,我可以给你提供以下的代码示例:
```python
# 导入所需的库
import jieba
import numpy as np
from PIL import Image
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
# 读取文本文件
text = open('text.txt', 'r', encoding='utf-8').read()
# 使用 jieba 分词
word_list = jieba.cut(text, cut_all=True)
# 将分词结果转换为字符串
words = ' '.join(word_list)
# 读取背景图片文件
mask = np.array(Image.open('mask.png'))
# 配置词云参数
wc = WordCloud(
background_color='white', # 背景颜色
mask=mask, # 背景图片
font_path='msyh.ttc', # 字体文件
max_words=200, # 最大词数
max_font_size=100, # 最大字体大小
width=800, # 图片宽度
height=800, # 图片高度
)
# 生成词云
wc.generate(words)
# 提取背景图片颜色
image_colors = ImageColorGenerator(mask)
# 显示词云
plt.imshow(wc)
plt.axis('off')
plt.show()
# 保存词云图片
wc.to_file('wordcloud.png')
```
这段代码使用了 jieba 进行中文分词,并使用 numpy、PIL、wordcloud 和 matplotlib 等库实现了词云绘制的功能。你可以将代码中的 `text.txt` 和 `mask.png` 分别替换为你需要绘制词云的文本和背景图片,然后运行代码即可生成词云图片。
阅读全文