소스 검색

Update accountServer: fix extend_dicts

StrayWarrior 8 달 전
부모
커밋
eb0aa17fca
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      routes/accountServer.py

+ 4 - 1
routes/accountServer.py

@@ -104,8 +104,11 @@ class AccountServer(object):
         )
         extend_dicts = {
             'view_count': good_df["show_view_count"].values.tolist(),
-            'view_count_rate': (good_df["show_view_count"] / good_df["view_count_avg"]).values.tolist()
         }
+        if 'view_count_avg' in good_df.columns:
+            extend_dicts['view_count_rate'] = \
+                (good_df["show_view_count"] / good_df["view_count_avg"]).values.tolist()
+
         account_interest = good_df["title"].values.tolist()
         print(account_interest)
         print(extend_dicts)