python生成词云图如何指定停词
时间: 2023-07-15 10:08:06 浏览: 147
python生成词云图
5星 · 资源好评率100%
要在生成词云图时指定停用词,你可以使用`WordCloud`类的`stopwords`参数。该参数接受一个停用词列表,表示不想在词云图中显示的单词。以下是修改后的代码示例:
```python
import pandas as pd
import matplotlib.pyplot as plt
from wordcloud import WordCloud
def create_wordcloud(city_name, stopwords):
data_path = r"C:\Users\17116\Desktop\大一\去哪儿.xlsx"
travel_data = pd.read_excel(data_path)
city_dataframe = travel_data.groupby('地点').get_group(city_name)
# 剔除玩法列中的"深度游"
city_dataframe = city_dataframe[city_dataframe['玩法'] != '深度游']
# 提取玩法列并合并为一个字符串
corpus = ' '.join(city_dataframe['玩法'])
# 创建词云对象并指定停用词
wordcloud = WordCloud(width=2000, height=500, font_path="C:/Windows/Fonts/simfang.ttf",
background_color='white', stopwords=stopwords).generate(corpus)
# 在显示词云图之前设置图像大小
plt.figure(figsize=(60, 30))
plt.imshow(wordcloud.recolor(), interpolation="bilinear")
plt.axis('off')
plt.show()
```
在上面的代码中,我们将停用词列表作为 `WordCloud` 类的 `stopwords` 参数传入,以指定不想在词云图中显示的单词。你可以在调用 `create_wordcloud` 函数时,将你自己定义的停用词列表作为参数传入。
阅读全文