Ошибка NullPointerException при попытке вызвать переданный метод

Просто выберите строку диапазона из df, например

row_count = df.shape[0]
split_point = int(row_count*1/5)
test_data, train_data = df[:split_point], df[split_point:]