python - 训练和测试拆分后的数据索引

Question

伙计们，我是数据科学和 Python 的新手。我正在研究回归问题。我的问题是，当我尝试绘制目标变量的测试部分时，我得到了奇怪的绘图类型

from sklearn.model_selection import train_test_split
train_input, test_input, train_target, test_target = 
train_test_split(features, target, test_size = 0.25, random_state = 42)
# Remove the labels from the dataset
plt.xlim(0,100)
plt.plot(test_target , 'g');

是因为附加到 test_target 的随机索引..？

我怎样才能得到这样的连续曲线

Answer 1

如果数据索引是问题，则使用：

df_train = df_train.reset_index()

如果您想重置并将其设置为 df 的另一列，请说“A”，然后执行以下操作：

df_train = df_train.reset_index().set_index('A')