从具有 IF 条件的 pandas 数据框生成 WordCloud

Generate WordCloud from a pandas dataframe with IF condition

我有一个 Pandas 数据框:

Text Sentiment
Great Positive
Very bad Negative

我正在尝试使用此代码绘制词云:

from wordcloud import WordCloud 
from matplotlib import colors
from nltk.corpus import stopwords
color_list=['#4d4d4d','#ffcc00','#000000',]
colormap=colors.ListedColormap(color_list)

words = ' '.join([Text for Text in df['Text']])
wordCloud = WordCloud(background_color='white',colormap=colormap, mode="RGB", width=2000 , height=1000).generate(words)
plt.figure(figsize=(20,10))
plt.imshow(wordCloud)
plt.show()

如何调整代码以根据列 'Sentiment' 的分类仅显示否定词?

创建单词时试试这个:

words = ' '.join([Text for Text in df[df['Sentiment']=='Negative']['Text']])