Попытка разбить фрейм данных на фреймы данных train, val и test на основе его индекса строки, например наблюдение 1 перейдет в тренировку, 2 - в val, а 3 - в тест, однако я наткнулся на препятствие. Вот мой код:
climbingTngDataset = pd.DataFrame([])
climbingValDataset = pd.DataFrame([])
climbingTestDataset = pd.DataFrame([])
for i in range(len(dfClimbing)):
if i % 2 == 0:
climbingValDataset.append(i)
if i % 3 == 0:
climbingTestDataset.append(i)
else:
climbingTngDataset.append(i)






Используйте groupby, чтобы разделить фрейм данных:
train, test, val = [
g for _, g in dfClimbing.groupby(dfClimbing.index % 3)
]
Демо
(С двумя разделениями вместо 3)
print(df)
Record ID Para Tag
0 1 A x
1 1 A y
2 2 B x
3 2 B y
4 1 A z
i, j = [g for _, g in df.groupby(df.index % 2)]
print(i)
Record ID Para Tag
0 1 A x
2 2 B x
4 1 A z
print(j)
Record ID Para Tag
1 1 A y
3 2 B y
Если на ваш вопрос был дан ответ, проголосуйте, пожалуйста, и отметьте наиболее полезный ответ как принятый. Вы можете принять ответ, щелкнув серую галочку слева от ответа, чтобы она стала зеленой. Спасибо.