Невидимый элемент на странице, который я не могу найти

Я бы использовал собственный Training_test_split scikit-learn и сгенерировал его из индекса

from sklearn.cross_validation import train_test_split


y = df.pop('output')
X = df

X_train,X_test,y_train,y_test = train_test_split(X.index,y,test_size=0.2)
X.iloc[X_train] # return dataframe train