equal samples train test split with pandas code example
Example 1: pandas split dataframe to train and test
train=df.sample(frac=0.8,random_state=200) #random state is a seed value
test=df.drop(train.index)
Example 2: sklearn train test split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)