Word Cloud in Python

Question

有一个数据帧：1

使用以下代码但得到ValueError：我们至少需要1个单词来绘制单词云，得到0。

有人知道如何解决这个问题吗；我正在尝试生成3个单词云——一个用于所有推文，一个用于真实推文，另一个用于虚假推文。

感谢您提前提供的帮助：(

df_real = df[df['label']==1]
df_fake = df[df['label']==0]
tweet_All = " ".join(review for review in df.Tweet)
tweet_real = " ".join(review for review in df_real.Tweet)
tweet_fake = " ".join(review for review in df_fake.Tweet)
fig, ax = plt.subplots(3, 1, figsize  = (30,30))
# Create and generate a word cloud image:
wordcloud_ALL = WordCloud(max_font_size=50, max_words=100, background_color="white").generate(tweet_All)
wordcloud_ADR = WordCloud(max_font_size=50, max_words=100, background_color="white").generate(str(tweet_real))
wordcloud_NADR = WordCloud(max_font_size=50, max_words=100, background_color="white").generate(str(tweet_fake))
# Display the generated image:
ax[0].imshow(wordcloud_ALL, interpolation='bilinear')
ax[0].set_title('All Tweets', fontsize=30)
ax[0].axis('off')
ax[1].imshow(wordcloud_ADR, interpolation='bilinear')
ax[1].set_title('Tweets under real Class',fontsize=30)
ax[1].axis('off')
ax[2].imshow(wordcloud_NADR, interpolation='bilinear')
ax[2].set_title('Tweets under fake Class',fontsize=30)
ax[2].axis('off')

Answer 1

df['label']的类型似乎是str(或object(：在您的屏幕截图中，它显示"假的"；作为条目。也许df_real和df_fake是空的？我会试着展示df_real和df_fake的头部。

一种解决方案可能是用取代第一条生产线

df_real = df[df['label']!='fake']
df_fake = df[df['label']=='fake']