分割数据集，但在训练数据集中无效

Question

我想完成这一部分，但是在train_data中，它没有用

dataset , h, c = re.load_csv("Irs.csv")
ratio = round(len(dataset)/5)
for i in range(0,5):
    Test_data = dataset[ratio*i:ratio*(i+1),:]
    Train_data = dataset[0:ratio*i&ratio*(i+1):-1,:]

我知道问题是&中的100％，但我不知道如何进行更改。

Answer 1

您可以构造构造Train_data所需的索引列表，然后使用.loc访问这些行，如下所示：

for i in range(0,5):
    Test_data = dataset.loc[ratio*i:ratio*(i+1),:]
    indices = list(range(0, ratio*i)) + list(range(ratio*(i+1), len(dataset)))
    Train_data = dataset.loc[indices,:]