lyrics_wordcloud_plot.py

# Para pegar o path do seu diretório de trabalho
path = os.getcwd()
# Definindo o nome da pasta que vai ser salvo os png
wordcloud_folder = 'wordcloud'

for artist in df.index:
    wordcloud = WordCloud(background_color="black",
                          max_words = 500,
                          width=1600, height=800).generate(df.loc[artist]['wordcloud'])
    
      # Caso queira plotar as imagens, descomentar essa seção
#     fig, ax = plt.subplots(figsize=(10,6))
#     ax.imshow(wordcloud, interpolation='bilinear')
#     ax.set_axis_off()
#     plt.imshow(wordcloud);

    # Criar uma pasta com o nome desejado caso não exista
    if not os.path.isdir(wordcloud_folder):
        os.mkdir(wordcloud_folder)
    
    # Salvar o arquivo com o nome do artista na pasta desejada
    else:
        wordcloud.to_file(os.path.join(path,wordcloud_folder,f"{artist}_wordcloud.png"))