liqian il y a 2 ans
Parent
commit
804ca4f83e
1 fichiers modifiés avec 1 ajouts et 2 suppressions
  1. 1 2
      rule_rank_h_18_19.py

+ 1 - 2
rule_rank_h_18_19.py

@@ -83,9 +83,8 @@ 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_videos = set(video_ids) - set(filtered_result)
     df['videoid'] = df['videoid'].astype(int)
-    filter_df = df[df['videoid'].isin(filter_videos)]
+    filter_df = df[df['videoid'].isin(filtered_result)]
     df = df.append(filter_df)
     df = df.drop_duplicates(['videoid'], keep=False)
     log_.info(f'filtered_df count = {len(df)}')