Train test split on index

Question

I have a dataset with a few columns, one being a ID and another representing a week. Each ID has the same amount of weeks, let\'s say 12.

I\'m using train_test_split