拆分数据集，但在训练数据集中不起作用

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,:]