spotify apiでrock playlist楽曲の特徴を取得して可視化

def get_playlist_tracks(df, sp):
    df_out = pd.DataFrame()
    
    for row in df.itertuples(index=False):
        items = sp.playlist_tracks(row[1], limit=100)['items']
        # merge playlist_id df&df_tracks
        df_tracks = df.merge(get_tracks(items, row[1]), on='playlist_id')
        
        # merge track_id df&df_features
        df_features = df_tracks.merge(get_audio_features(df_tracks, sp), on='track_id')
        
        df_out = df_out.append(df_features)
    return df_out
    
def get_tracks(items, playlist_id):
    #for item in res_tracks['items']:
    l = []

    for item in items:
        track = item['track']
    
        track_id = track['id']
        artist_name = track['artists'][0]['name']
        track_name = track['name']
        release_date = track['album']['release_date']
        track_popularity = track['popularity']
        
        l.append([playlist_id, track_id, artist_name
                  ,track_name, release_date, track_popularity])
        
        time.sleep(1)
        
    cols = ['playlist_id', 'track_id', 'artist_name', 'track_name'
            ,'release_date', 'track_popularity'] 
    df = pd.DataFrame(l, columns=cols)
    
    
        #print(track)
    return df
    
# 50ごとに取得
def get_audio_features(df, sp, limit=50):
    df_features = pd.DataFrame()
    for i in range(0, int(len(df) / limit)+1):
        tracks = list(df['track_id'].iloc[i*limit:i*limit+limit])
        #print('track len', len(df))
        #print(f'range: {i*limit}~{i*limit+limit}')
        df_features = df_features.append(pd.DataFrame(sp.audio_features(tracks=tracks)))

        time.sleep(1)
    
    df_features.rename(columns={'id': 'track_id'}, inplace=True)
    #print('len df_feat', len(df_features))
    #print('df_feat cols', df_features.columns)

    return df_features
    

import pandas as pd
import time
import spotipy

# init spotify api
client_id = '240xxxxx'
client_secret = 'a62xxxxxxxx'
client_credentials_manager = spotipy.oauth2.SpotifyClientCredentials(client_id, client_secret)

sp = spotipy.Spotify(client_credentials_manager=client_credentials_manager)

# playlists
playlist_cols = ['title', 'playlist_id']
playlists = [['70s Rock Anthems', '37i9dQZF1DWWwzidNQX6jx']
            ,['Women of Rock', '37i9dQZF1DXd0ZFXhY0CRF']
            ,['80s Rock Anthems', '37i9dQZF1DX1spT6G94GFC']
            ,['60s Rock Anthems', '37i9dQZF1DWWzBc3TOlaAV']
            ,['90s Rock Anthems', '37i9dQZF1DX1rVvRgjX59F']
            ,['00s Rock Anthems', '37i9dQZF1DX3oM43CtKnRV']
            ]
df_playlist = pd.DataFrame(playlists, columns=playlist_cols)

# 検索してplaylistの詳細情報を追加
%time df_all = get_playlist_tracks(df_playlist, sp

df_all.to_csv('./spotify_playlists.csv', index=False)

# 検索
query = "10s"
results = sp.search(q=query, limit=20, type='playlist', offset=1)
for idx, track in enumerate(results['playlists']['items']):
    print(idx, track['owner']['display_name'],  track['id'], track['name']
          ,  track['tracks']['total']
          , track['external_urls']['spotify']
         )
    print(track['description'])