引入txt并制作词云,可以用你自己想要的形状
时间: 2024-03-11 18:48:13 浏览: 91
Python基于WordCloud制作词云图
要制作词云,你可以使用Python中的第三方库wordcloud,它可以很方便地生成词云图。下面是一个示例代码,假设你要制作一个心形词云:
```python
import os
import numpy as np
from PIL import Image
from wordcloud import WordCloud, STOPWORDS
# 设置词云形状的图片路径
shape_path = "heart.png"
# 读取要生成词云的文本文件
txt_path = "sample.txt"
with open(txt_path, "r", encoding="utf-8") as f:
text = f.read()
# 加载停用词
stopwords = set(STOPWORDS)
# 加载形状图片
shape = np.array(Image.open(shape_path))
# 设置词云参数
wc = WordCloud(background_color="white", max_words=1000, mask=shape, stopwords=stopwords, contour_width=3, contour_color='steelblue')
# 生成词云
wc.generate(text)
# 保存词云图片
output_dir = "output"
if not os.path.exists(output_dir):
os.mkdir(output_dir)
wc.to_file(os.path.join(output_dir, "heart_wordcloud.png"))
```
这个代码会读取指定路径下的文本文件,并以心形图片为形状生成词云。你可以根据需要修改路径和参数,制作自己想要的形状和风格的词云。注意,为了生成指定形状的词云,你需要提供一张对应形状的图片作为mask参数。
阅读全文