pandas 从总样本中选取一定数量的正负样本,并合并保存

data_ = pd.read()

# 从总样本中抽取可复现的随机抽取60个负样本,负样本即为[label]=0的数据
negative_data = data_[data_['label']==0].sample(n=60,random_state=1)

# 从总样本中抽取全部正样本,正样本即为[label]=1的数据
positive_data = data_[data_['label']==1]

# 将正负样本合在一起
processing_data = negative_data.append(positive_data)

# 保存这个数据为excel,不加索引
processing_data.to_excel('./FE_out.xlsx',index=False)

猜你喜欢

转载自blog.csdn.net/qq_56039091/article/details/123930635
今日推荐