Saya menggunakan fungsi yang sangat sederhana untuk mengatur waktu bagian dari eksekusi kode:
import time
def timing():
start_time = time.time()
return lambda x: print("[{:.2f}s] {}".format(time.time() - start_time, x))
Dan untuk menggunakannya, panggil saja sebelum kode untuk mengukur untuk mengambil waktu fungsi, dan kemudian memanggil fungsi setelah kode dengan komentar. Waktu akan muncul di depan komentar. Sebagai contoh:
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)))
Maka hasilnya akan terlihat seperti ini:
[9.35s] Loaded 1458644 rows data from 'train'