Как построить Wordlcoud для каждого столбца в Pandas Dataframe? - PullRequest
Новая
       19

Как построить Wordlcoud для каждого столбца в Pandas Dataframe?

0 голосов
/ 29 июня 2018

У меня есть пандас с N строками.

В каждой строке N много слов.

Я пытаюсь построить wordloud для каждого столбца.

ДФ: 

    prediction                    text
0           1  [menu, food, food and chicken...
1           2  [transfer, money, transfer, transfer, trans...
2           0  [cold,cold,cold,hot,love...

код 

import matplotlib.pyplot as plt
from wordcloud import WordCloud, STOPWORDS

def show_wordcloud(data, title):

        wordcloud = WordCloud(
        background_color='white',
        max_words=200,
        max_font_size=40,
        scale=3,
    ).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.savefig('cluster{}.png'.format(title))
    plt.show()


for row in df.iterrows():
    show_wordcloud(row['text'], row["prediction"])

Но я получаю пустые изображения для всех столбцов, кроме последнего.

Как мне это исправить?

Спасибо

...