ActorDirector
表:
+-------------+---------+ | Column Name | Type | +-------------+---------+ | actor_id | int | | director_id | int | | timestamp | int | +-------------+---------+ timestamp 是这张表的主键(具有唯一值的列).
编写解决方案找出合作过至少三次的演员和导演的 id 对 (actor_id, director_id)
import pandas as pd
def actors_and_directors(actor_director: pd.DataFrame) -> pd.DataFrame:
actor_director_group = actor_director.groupby(['actor_id', 'director_id']).size().reset_index(name='cnt')
actor_director_group = actor_director_group[actor_director_group['cnt']>=3][['actor_id', 'director_id']]
#print(actor_director_group)
return actor_director_group