J'ai utilisé une fonction très simple pour chronométrer une partie de l'exécution du code:
import time
def timing():
start_time = time.time()
return lambda x: print("[{:.2f}s] {}".format(time.time() - start_time, x))
Et pour l'utiliser, il suffit de l'appeler avant le code à mesurer pour récupérer le timing de la fonction, puis d'appeler la fonction après le code avec des commentaires. L'heure apparaîtra devant les commentaires. Par exemple:
t = timing()
train = pd.read_csv('train.csv',
dtype={
'id': str,
'vendor_id': str,
'pickup_datetime': str,
'dropoff_datetime': str,
'passenger_count': int,
'pickup_longitude': np.float64,
'pickup_latitude': np.float64,
'dropoff_longitude': np.float64,
'dropoff_latitude': np.float64,
'store_and_fwd_flag': str,
'trip_duration': int,
},
parse_dates = ['pickup_datetime', 'dropoff_datetime'],
)
t("Loaded {} rows data from 'train'".format(len(train)))
Ensuite, la sortie ressemblera à ceci:
[9.35s] Loaded 1458644 rows data from 'train'