example split data

pandas.py

import pandas as pd
from sklearn.model_selection import train_test_split


# import my data into a dataframe (df)
df = pd.read_csv("mydata.csv")

# random split 60/40% solution / public dataframe
solution, public_df = train_test_split(df,test_size=0.40,random_state=101)

solution.to_csv("solution.csv", index=False)
public_df.to_csv("public.csv", index=False)