Splitting train and test data by a particular variable

Question

i am trying this code for splitting data into train and test for a logistic regression:

"""

from sklearn.model_selection import train_test_s