Случайная выборка Pandas также будет работать
train=df.sample(frac=0.8,random_state=200) test=df.drop(train.index)