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])
本文来自博客园,作者:赫凯,转载请注明原文链接:https://www.cnblogs.com/heKaiii/p/17137376.html

浙公网安备 33010602011771号