Séparer les données (train, test) : même valeur même groupe

Bonjour,
je souhaiterais savoir s'il existe un moyen de séparer des données en deux (train et test), mais avec une petite subtilité. C'est a dire qu'on a dans nos données une variable ID ( plusieurs lignes peuvent avoir le même id), lorsqu'une ligne avec un id = 3 par exemple a été mise dans le train alors toutes les lignes ayant cet id doivent être dans le train.
Il faut aussi avoir 70% des données dans le train et 30 dans le test.

Exemple:
list= [1,2,3,1,2,3,4]
list1=[1,1,2,2,4]
list2=[3,3]

Merci d'avance pour toute aide