Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 35 additions & 0 deletions Problem1_GroupsoldByDate.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
import pandas as pd

#Soln 1 Without pandas
# def categorize_products(activities: pd.DataFrame) -> pd.DataFrame:
# dict={}

# for i in range(len(activities)):
# sell_date=activities['sell_date'][i]
# product=activities['product'][i]

# if sell_date not in dict:
# dict[sell_date]=set()
# dict[sell_date].add(product)


# result=[]

# for key,value in dict.items():
# temp=[]
# for product in value:
# temp.append(product)
# temp.sort()
# s=','.join(temp)
# result.append([key,len(value),s])

# df=pd.DataFrame(result,columns=['sell_date','num_sold','products'])
# return df.sort_values(by='sell_date')

#Soln 2 With pandas
def categorize_products(activities: pd.DataFrame) -> pd.DataFrame:
groups=activities.groupby('sell_date').agg(
num_sold=('product','nunique'),
products=('product',lambda x: ','.join(sorted(set (x))))
).reset_index()
return groups.sort_values(by='sell_date')
8 changes: 8 additions & 0 deletions Problem2_DailyLeadsPartners.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
import pandas as pd

def daily_leads_and_partners(daily_sales: pd.DataFrame) -> pd.DataFrame:
df= daily_sales.groupby(['date_id','make_name']).agg(
unique_leads=('lead_id','nunique'),
unique_partners=('partner_id','nunique')
).reset_index()
return df
6 changes: 6 additions & 0 deletions Problem3_ActorDirector.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
import pandas as pd

def actors_and_directors(actor_director: pd.DataFrame) -> pd.DataFrame:
df=actor_director.groupby(['actor_id','director_id'])['timestamp'].size().reset_index(name='count')
df=df[df['count']>=3]
return df[['actor_id','director_id']] # type: ignore