liqian 2 年 前
コミット
cb43173b10
1 ファイル変更2 行追加1 行削除
  1. 2 1
      rule_rank_h_18_19.py

+ 2 - 1
rule_rank_h_18_19.py

@@ -83,8 +83,9 @@ def cal_score(df):
     log_.info(f'initial_df count = {len(df)}')
     video_ids = [int(video_id) for video_id in df['videoid']]
     filtered_result = filter_video_status(video_ids=video_ids)
+    filter_result = set(video_ids) - set(filtered_result)
     df['videoid'] = df['videoid'].astype(int)
-    filter_df = df[df['videoid'].isin(filtered_result)]
+    filter_df = df[df['videoid'].isin(filter_result)]
     df = df.append(filter_df)
     df = df.drop_duplicates(['videoid'], keep=False)
     log_.info(f'filtered_df count = {len(df)}')