1148.Article Views I

My code

import pandas as pd

def article_views(views: pd.DataFrame) -> pd.DataFrame:
    viewer_df =  views[(views['author_id']) == (views['viewer_id'])]
    unique_authors = sorted(viewer_df['author_id'].unique())
    ans = pd.DataFrame({'id': unique_authors})
    return ans

Best Code

import pandas as pd

def article_views(views: pd.DataFrame) -> pd.DataFrame:
  return views.loc[views["author_id"] == views["viewer_id"], ["author_id"]].drop_duplicates().rename(columns={"author_id" : "id"}).sort_values(by=["id"])