word-cloud-most-common.py

# Word cloud with most common words
def show_wordcloud(text, artist):
    # Create and generate a word cloud image:
    wordcloud = WordCloud(stopwords=stoplist, background_color="white").generate(text)

    # Display the generated image:
    fig = plt.figure(figsize=(25,10))
    plt.imshow(wordcloud, interpolation='bilinear')
    plt.title(f'Word Cloud for {artist}', fontsize=20)
    plt.axis("off")
    plt.show()
    fig.savefig(f'word_cloud_{artist}.png', format='png', dpi=500)
    
def get_wordcloud(df, artist):
    dataframe = df[df['artist'] == artist]
    # Get all texts and generate a cloud
    text = " ".join(review for review in dataframe.lyric)
    show_wordcloud(text, artist)