split into train and test pandas code example
Example 1: train test split pandas
from sklearn.model_selection import train_test_split
train, test = train_test_split(df, test_size=0.2)
Example 2: pandas split dataframe to train and test
train=df.sample(frac=0.8,random_state=200)
test=df.drop(train.index)
Example 3: split data train, test by id python
train_inds, test_inds = next(GroupShuffleSplit(test_size=.20, n_splits=2, random_state = 7).split(df, groups=df['Group_Id']))
train = df.iloc[train_inds]
test = df.iloc[test_inds]