Python有特定形状的词云写出代码
时间: 2023-12-14 21:40:18 浏览: 74
Python制作词云图代码
是的,Python中有一个叫做WordCloud的库可以用来生成词云。下面是一个简单的例子,可以生成心形的词云:
```python
from wordcloud import WordCloud
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image
# 读取文本文件
text = open("mytext.txt").read()
# 读取心形图案
heart_mask = np.array(Image.open("heart.png"))
# 创建词云对象
wc = WordCloud(background_color="white", max_words=2000, mask=heart_mask, contour_width=3, contour_color='pink')
# 生成词云
wc.generate(text)
# 显示词云
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.show()
# 保存词云
wc.to_file("heart_wordcloud.png")
```
其中,`mytext.txt`是要生成词云的文本文件,`heart.png`是心形图案的图片文件。你可以根据需要替换这两个文件。在上述代码中,我们使用了`WordCloud`的`mask`参数来指定词云的形状,从而生成了一个心形的词云。
阅读全文