How to create a word cloud from a corpus in Python?

后端 未结 5 1645
刺人心
刺人心 2020-12-04 09:57

From Creating a subset of words from a corpus in R, the answerer can easily convert a term-document matrix into a word cloud easily.

Is there a similar

5条回答
  •  时光取名叫无心
    2020-12-04 10:50

    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(Samsung_Reviews_Negative['Reviews'])
    show_wordcloud(Samsung_Reviews_positive['Reviews'])
    

提交回复
热议问题