add

make_data.py

@@ -46,7 +46,12 @@ def doctor():
     df["commission"] = (df["doctor_ad_money_30_days"] + df["budan_payment_30_days"])/df["tmp"]
     df["pv_ad"] = df["expand_rechange_amount_30"]/df["tmp"]
     df["score"] = df["ctr"]**0.5 * (df["commission"] + df["pv_ad"])
-    print(df)
+    columns = ["score","ctr","commission","pv_ad","service_exposure_pv_30","service_ctr_30","expert_exposure_pv_30","expert_pv_30",
+            "merchant_id","doctor_ad_money_30_days","expand_rechange_amount_30","service_pv_30",
+            "mexpert_pv_30","organization_pv_30","budan_payment_30_days"]
+    data = df.loc[:, columns]
+    print(data)
+    data.to_csv('/home/gmuser/score.csv',index=False)
 
 
 if __name__ == "__main__":