使用python绘制词云图,使用背景图片生成词云图
时间: 2023-10-24 14:11:21 浏览: 124
使用Python制作的词云图
5星 · 资源好评率100%
要使用Python绘制词云图,可以使用Python第三方库`wordcloud`。这个库可以根据给定的文本数据生成词云图。要生成带有背景图片的词云图,可以使用`WordCloud`类的`mask`参数。
以下是一个示例代码,演示如何使用`wordcloud`库生成带有背景图片的词云图:
```python
from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np
# 读取文本文件
text = open('text.txt', 'r', encoding='utf-8').read()
# 读取背景图片
background = np.array(Image.open("background.png"))
# 创建停用词集合
stopwords = set(STOPWORDS)
stopwords.add("said")
# 创建词云对象
wc = WordCloud(background_color="white", max_words=2000, mask=background,
stopwords=stopwords, contour_width=3, contour_color='steelblue')
# 生成词云图
wc.generate(text)
# 显示词云图
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.show()
```
在上面的代码中,首先使用`open`函数读取文本文件,然后使用`np.array`函数读取背景图片。接着创建停用词集合,用于过滤掉一些常见的词语,比如“said”。然后创建`WordCloud`对象,设置一些参数,比如背景颜色、最大词数、掩膜图片、停用词等。最后使用`generate`方法生成词云图,使用`imshow`方法显示词云图。
注意,该代码中使用了词云图的默认字体。如果你想更改字体,可以在创建`WordCloud`对象时使用`font_path`参数,指定你想要使用的字体文件路径。
阅读全文