diff --git a/problem1.py b/problem1.py new file mode 100644 index 0000000..baef63d --- /dev/null +++ b/problem1.py @@ -0,0 +1,9 @@ +import pandas as pd + +def article_views(views: pd.DataFrame) -> pd.DataFrame: + condition = views[views['author_id'] == views['viewer_id']].drop_duplicates(subset=['author_id']) + condition = condition.rename(columns={'author_id': 'id'}) + + result = condition['id'].sort_values() + + return pd.DataFrame(result) \ No newline at end of file diff --git a/problem2.py b/problem2.py new file mode 100644 index 0000000..6540b12 --- /dev/null +++ b/problem2.py @@ -0,0 +1,5 @@ +import pandas as pd + +def invalid_tweets(tweets: pd.DataFrame) -> pd.DataFrame: + invalid_tweets_df = tweets[tweets['content'].str.len() > 15] + return invalid_tweets_df[['tweet_id']] \ No newline at end of file