df1_1 = df1[df1.target1]
df1_0 = df1[df1.target0]
df_1 = pd.DataFrame()
df_0 = pd.DataFrame()
for dt in df1.date.unique():
qx_1 = df1_1[df1_1.date==dt]
if len(qx_1)<1000:
qx1 = qx_1[qx_1.date == dt]
else:
qx1 = qx_1[qx_1.date == dt].sample(n=1000)
df_1 = pd.concat([df_1,qx1],axis=0)
for dt in df1.date.unique():
qx_0 = df1_0[df1_0.date==dt]
if len(qx_0)<3000:
qx0 = qx_0[qx_0.date == dt]
else:
qx0 = qx_0[qx_0.date == dt].sample(n=3000)
df_0 = pd.concat([df_0,qx0],axis=0)
df1 = pd.concat([df_1,df_0],axis=0)