liqian 2 년 전
부모
커밋
51521bd6dc
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)}')