.

eda/smart_rank/test_divided_portrait.py

@@ -44,6 +44,7 @@ def user_portrait_action_statistic(cl_id):
             ud8_list = ud[8].split("，")
 
             if ud[0] in ud_dict.keys():
+                if ud[1] in ud_dict[ud[0]].keys():
                     ud_dict[ud[0]][ud[1]]["first_solutions"].extend(ud2_list)
                     ud_dict[ud[0]][ud[1]]["second_solutions"].extend(ud3_list)
                     ud_dict[ud[0]][ud[1]]["first_demands"].extend(ud4_list)
@@ -51,6 +52,15 @@ def user_portrait_action_statistic(cl_id):
                     ud_dict[ud[0]][ud[1]]["first_positions"].extend(ud6_list)
                     ud_dict[ud[0]][ud[1]]["second_positions"].extend(ud7_list)
                     ud_dict[ud[0]][ud[1]]["projects"].extend(ud8_list)
+                else:
+                    ud_dict[ud[0]][ud[1]]["first_solutions"] = ud2_list
+                    ud_dict[ud[0]][ud[1]]["second_solutions"] = ud3_list
+                    ud_dict[ud[0]][ud[1]]["first_demands"] = ud4_list
+                    ud_dict[ud[0]][ud[1]]["second_demands"] = ud5_list
+                    ud_dict[ud[0]][ud[1]]["first_positions"] = ud6_list
+                    ud_dict[ud[0]][ud[1]]["second_positions"] =ud7_list
+                    ud_dict[ud[0]][ud[1]]["projects"] = ud8_list
+
             else:
                 ud_dict[ud[0]] = {}
                 ud_dict[ud[0]][ud[1]] = {"first_solutions": ud2_list,"second_solutions": ud3_list,
@@ -94,7 +104,7 @@ if __name__ == '__main__':
     second_solutions = []
     projects = []
 
-    # cl_ids = [1, 3]
+    cl_ids = [1, 3]
     data = user_portrait_action_statistic(cl_ids)
 
     for key, values in data.items():

eda/smart_rank/tool.py

@@ -474,6 +474,7 @@ def user_portrait_action_statistic(cl_id):
             ud8_list = ud[8].split("，")
 
             if ud[0] in ud_dict.keys():
+                if ud[1] in ud_dict[ud[0]].keys():
                     ud_dict[ud[0]][ud[1]]["first_solutions"].extend(ud2_list)
                     ud_dict[ud[0]][ud[1]]["second_solutions"].extend(ud3_list)
                     ud_dict[ud[0]][ud[1]]["first_demands"].extend(ud4_list)
@@ -481,6 +482,14 @@ def user_portrait_action_statistic(cl_id):
                     ud_dict[ud[0]][ud[1]]["first_positions"].extend(ud6_list)
                     ud_dict[ud[0]][ud[1]]["second_positions"].extend(ud7_list)
                     ud_dict[ud[0]][ud[1]]["projects"].extend(ud8_list)
+                else:
+                    ud_dict[ud[0]][ud[1]]["first_solutions"] = ud2_list
+                    ud_dict[ud[0]][ud[1]]["second_solutions"] = ud3_list
+                    ud_dict[ud[0]][ud[1]]["first_demands"] = ud4_list
+                    ud_dict[ud[0]][ud[1]]["second_demands"] = ud5_list
+                    ud_dict[ud[0]][ud[1]]["first_positions"] = ud6_list
+                    ud_dict[ud[0]][ud[1]]["second_positions"] = ud7_list
+                    ud_dict[ud[0]][ud[1]]["projects"] = ud8_list
             else:
                 ud_dict[ud[0]] = {}
                 ud_dict[ud[0]][ud[1]] = {"first_solutions": ud2_list,"second_solutions": ud3_list,