python - stratify - Sklearn train_test_split; reteniendo valores únicos de la(s) columna(s) en el co