Python Pandas里筛选剩下的行

在Pandas会做训练数据和测试数据,提取数据时一般是随机抽样,抽样过后需要把剩下数据归到测试里,可以这么写

# data_df就是全部数据
train_df = data_df.sample(frac=0.9, axis=0, random_state=2022)
test_df = data_df[~data_df.index.isin(train_df.index)]
posted @ 2022-12-14 09:30  赫凯  阅读(30)  评论(0)    收藏  举报