python - 尝试拆分为训练/测试集时未找到 X 索引

Question

我有一个大型 float 数据帧X和一个目标值数组y，我试图将其拆分为训练集和测试集。但是我收到以下错误:
KeyError:“没有 [Int64Index([ 0, 1, 2, 3, 5, 6, 7, 9, 10, 11, 13, 14, 15, 16, 17, 18, 19,\n 20, 21, 22 , 23, 26, 27, 28, 29, 30, 31, 32, 33, 35, 38, 40, 41,\n 42],\n dtype='int64')] 位于 [列]"


这是我的代码:

split = KFold(n_splits=5, shuffle=True, random_state=42)
for train_index, test_index in split.split(X, y):
    X_train, X_test = X[train_index], X[test_index]
    y_train, y_test = y[train_index], y[test_index]

我尝试使用以下方法将 df 索引转换为 int:

X = X.set_index(X.index.astype(int))

但错误仍然相同。我认为 df 的索引发生了一些问题，这意味着代码无法找到训练/测试索引，但我不明白为什么会这样？

Answer 1

如果 X 和 y 是 numpy 数组而不是数据帧，则您的代码可以工作。要与 pandas 数据框一起使用，请将 X[train_index] 更改为 X.iloc[train_index]