将df格式转化为字典

predictDiary.py

@@ -65,16 +65,15 @@ def upload_predict(user_profile, instance):
                                 "result/{0}_output.txt".format(user_profile['device_id']), header=None)
     probabilities = probabilities.rename(columns={0: "prob"})
     probabilities["cid"] = instance['cid']
-    probabilities = probabilities.sort_values(by="0",ascending=False)
+    probabilities = probabilities.sort_values(by="0", ascending=False)
 
-    wrapper_result(probabilities,user_profile['device_id'])
+    wrapper_result(probabilities, user_profile['device_id'])
 
 
-def wrapper_result(prob,device_id):
+def wrapper_result(prob, device_id):
     prob = prob.head(500)
-    prob["url"] = prob["cid"].apply(lambda x:"http://m.igengmei.com/diary_book/" + prob[x.index('|') + 1:] + '/')
-    prob.to_csv(DIRECTORY_PATH+"result/{}_feed".format(device_id))
-
+    prob["url"] = prob["cid"].apply(lambda x: "http://m.igengmei.com/diary_book/" + prob[x.index('|') + 1:] + '/')
+    prob.to_csv(DIRECTORY_PATH + "result/{}_feed".format(device_id))
 
 
 def router(device_id):
@@ -82,7 +81,6 @@ def router(device_id):
 
     if is_exist:
         predict(user_profile)
-
     else:
         print('Sorry, we don\'t have you.')
 
@@ -90,4 +88,3 @@ def router(device_id):
 if __name__ == "__main__":
     router(device_id='358035085192742')
 
-# 预测一些真实的device_id

userProfile.py

@@ -6,5 +6,11 @@ def fetch_user_profile(device_id):
     sql = "select device_id,city_id from " \
           "data_feed_click where device_id = '{0}' limit 1".format(device_id)
     user_profile = con_sql(sql)
+    if user_profile.empty:
+        return {}, user_profile.empty
+
     user_profile = user_profile.rename(columns={0:"device_id",1:"city_id"})
-    return user_profile, not user_profile.empty
+    user_profile_dict = {}
+    for i in user_profile.columns:
+        user_profile_dict[i] = user_profile.loc[0, i]
+    return user_profile_dict, not user_profile.empty