define inspect data function

def inspect_data(df, col=None, n_rows=5):
    print(f'data shape: {df.shape}')
    if col is None:
        col = df.columns
    display(df[col].head(n_rows))