df=df.reset_index(drop=True)
============
df = pd.read_csv('./train_file/train.csv').dropna()
df_test = df.sample(frac=0.2, random_state=123456)
df_train = df.drop(df_test.index)
df=df.reset_index(drop=True)
============
df = pd.read_csv('./train_file/train.csv').dropna()
df_test = df.sample(frac=0.2, random_state=123456)
df_train = df.drop(df_test.index)
转载于:https://www.cnblogs.com/yjybupt/p/10727198.html