KeyError：“ [Int64Index（[2，3]，dtype ='int64'）]都不在[列]中”

Question

以下是我遇到错误的代码的简化版本： KeyError：“ [Int64Index（[2，3]，dtype ='int64'）]都不在[列]中”

到目前为止，我已经浏览了this和其他相关文章；但是解决方案不起作用。

这里的错误是什么以及如何解决？

import numpy as np
import pandas as pd
from sklearn.model_selection import KFold

df = pd.DataFrame({'num_legs': [2, 4, 8, 0], 'num_wings': [2, 0, 0, 0], 'num_specimen_seen': [10, 2, 1, 8]}, index=[0,1,2,3])

kf = KFold(n_splits=2, shuffle=True)
for train_index, val_index in kf.split(df):
    df_train, df_val = df[train_index], df[val_index]