类似股票的多路数据集，如何划分各个股票的最后一天的数据作为测试集

我将所有股票数据合到一个excel里面，并将group_id传入df的code列，希望训练是以前5天数据预测后一天的，每支股票只需要留下最后一天的数据作为验证数据。这个要如何设置？