Pandas 对行数据过采样以及抽样

# target等于1的行扩大20倍
df1 = pd.DataFrame(np.repeat(df[df['target']==1].values, 20, axis=0), columns=df.columns)
# target等于2的行抽样4000个
df2 = df[df['target']==0].sample(n=4000, random_state=1)
# 合并
df= pd.concat([df1, df2])
posted @ 2022-09-28 14:27  赫凯  阅读(66)  评论(0)    收藏  举报