在python中,显示词云怎么设置背景图片
时间: 2024-05-03 18:16:45 浏览: 57
在 Python 中,可以使用 `wordcloud` 库来生成词云,并且可以设置背景图片。具体步骤如下:
1. 安装 `wordcloud` 库:在命令行中输入 `pip install wordcloud`。
2. 导入库并读取文本文件:在 Python 脚本中导入 `wordcloud` 库,并读取文本文件。
```python
from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
text = open('example.txt').read()
```
3. 设置背景图片:使用 `WordCloud()` 函数的 `mask` 参数指定背景图片。
```python
background = plt.imread('background.png')
wordcloud = WordCloud(stopwords=STOPWORDS, background_color='white', mask=background).generate(text)
```
其中,`background_color` 参数可以设置背景颜色,`stopwords` 参数可以设置停用词(即不在词云中显示的词语)。
4. 显示词云:使用 `imshow()` 函数显示词云。
```python
plt.imshow(wordcloud)
plt.axis('off')
plt.show()
```
完整的代码示例:
```python
from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
text = open('example.txt').read()
background = plt.imread('background.png')
wordcloud = WordCloud(stopwords=STOPWORDS, background_color='white', mask=background).generate(text)
plt.imshow(wordcloud)
plt.axis('off')
plt.show()
```
阅读全文