liqian 3 år sedan
förälder
incheckning
65216b58be
1 ändrade filer med 2 tillägg och 1 borttagningar
  1. 2 1
      rule_rank_h_18_19.py

+ 2 - 1
rule_rank_h_18_19.py

@@ -84,7 +84,8 @@ def cal_score(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)
-    filter_df = df[int(df['videoid'].values).isin(filter_videos)]
+    df['videoid'] = df['videoid'].astype(int)
+    filter_df = df[df['videoid'].isin(filter_videos)]
     df = df.append(filter_df)
     df = df.drop_duplicates(['videoid'], keep=False)
     log_.info(f'filtered_df count = {len(df)}')