时间戳

首页 > 解决方案 > 如何在 Python 中创建不同组的词云？

python - 如何在 Python 中创建不同组的词云？

问题描述

我有以下数据，我正在尝试为每个“人”（即 1、2、3）创建 wordcloud。我正在使用以下代码创建词云，但我不确定如何为每个组创建词云。注意：我是 python 新手，如果需要，请提供解释和参考。

数据集：

import pandas as pd
data = {'Person':['1', '1','1','2','2','2','2','3','3'],'Response':['I like to eat','You have nice day','My name is ','I like to eat','You have nice day','My name is','This is it','I like to eat','You have nice day'],
       }
df = pd.DataFrame(data)

Wordcloud 的代码

from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
stopwords = set(STOPWORDS)

def show_wordcloud(data, title = None):
    wordcloud = WordCloud(
        background_color='white',
        stopwords=stopwords,
        max_words=200,
        max_font_size=40, 
        scale=3,
        random_state=1 # chosen at random by flipping a coin; it was heads
    ).generate(str(data))

    fig = plt.figure(1, figsize=(12, 12))
    plt.axis('off')
    if title: 
        fig.suptitle(title, fontsize=20)
        fig.subplots_adjust(top=2.3)

    plt.imshow(wordcloud)
    plt.show()

show_wordcloud(data['Response'])

标签： pythonpython-3.xword-cloud

解决方案

这应该可以工作，但是添加了一个我不确定的额外输出：

df.groupby('Person').apply(
    lambda x: show_wordcloud(x.Response.tolist(), title=f"Person {x.name}")
)

推荐阅读