df_permutated = df.sample(frac=1) train_size = 0.8 train_end = int(len(df_permutated)*train_size) df_train = df_permutated[:train_end] df_test = df_permutated[train_end:]