Actors and Directors Who Cooperated At Least Three Times


Źródło
select actor_id, director_id
from ActorDirector 
GROUP BY actor_id, director_id
HAVING COUNT(*) >= 3;
def actors_and_directors(actor_director: pd.DataFrame) -> pd.DataFrame:
    # df = actor_director.groupby(["actor_id", "director_id"]).count().reset_index()
    # return df[df["timestamp"] >= 3][["actor_id", "director_id"]]

    df = (
        actor_director.groupby(["actor_id", "director_id"])
        .size()
        .reset_index(name="count")
    )
    print(df)
    return df[df["count"] >= 3][["actor_id", "director_id"]];

Dodaj komentarz

Twój adres e-mail nie zostanie opublikowany. Wymagane pola są oznaczone *