|
|
@@ -4003,9 +4003,9 @@ class DistrictPredictor():
|
|
|
df = pd.DataFrame(score_l, columns=['id', 'score', 'province', 'city', 'district'])
|
|
|
df_pro = df.groupby('province').sum().sort_values(by=['score'], ascending=False)
|
|
|
pro_id = df_pro.index[0]
|
|
|
- # if df_pro.loc[pro_id, 'score'] < 0.1: # 省级评分小于0.1的不要
|
|
|
- # print('评分低于0.1', df_pro.loc[pro_id, 'score'], self.dist_dic[pro_id]['地区'])
|
|
|
- # return area_dic
|
|
|
+ if df_pro.loc[pro_id, 'score'] < 0.1: # 省级评分小于0.1的不要
|
|
|
+ # print('评分低于0.1', df_pro.loc[pro_id, 'score'], self.dist_dic[pro_id]['地区'])
|
|
|
+ return {'district': area_dic}
|
|
|
area_dic['province'] = self.dist_dic[pro_id]['地区']
|
|
|
area_dic['area'] = self.dist_dic[pro_id]['大区']
|
|
|
df = df[df['city'] != ""]
|
