Top Albums

spotifyWrapped.py

topAlbumsDf = pd.DataFrame(columns = ['Rank', 'Album', 'Streams'])
topAlbums = []
albumList = data['Album'].tolist()
for i in range(50): #change to see different number of top albums
    def most_frequent(albumList):
        return max(set(albumList), key = albumList.count)
    topAlbum = most_frequent(albumList)
    topAlbumsDf = topAlbumsDf.append({'Rank': i+1, 'Album': topAlbum, 
        'Streams': albumList.count(topAlbum)}, ignore_index = True)
    topAlbums.append(topAlbum)
    albumList = [i for i in albumList if i != topAlbum]
display(topAlbumsDf.to_string(index = False))
topAlbumsDf.to_excel(f'{outputPath}/spotify_top_albums.xlsx', index = False)