Задача: Напишите решение, чтобы найти все пары, (actor_id, director_id) в которых актер сотрудничал с режиссером по крайней мере три раза.
Дано: timestamp - это первичный ключ (столбец с уникальными значениями) для этой таблицы.
Решение:
def actors_and_directors(actor_director: pd.DataFrame) -> pd.DataFrame: stats = actor_director.groupby( ['actor_id', 'director_id'], ).agg( count=('director_id', 'count'), ).reset_index() return stats[stats['count'] >= 3][['actor_id', 'director_id']]
Результат: Источник: https://leetcode.com/problems/actors-and-directors-who-cooperated-at-least-three-times/description/
Задача: Напишите решение, чтобы найти все пары, (actor_id, director_id) в которых актер сотрудничал с режиссером по крайней мере три раза.
Дано:
timestamp - это первичный ключ (столбец с уникальными значениями) для этой таблицы.
Решение:
def actors_and_directors(actor_director: pd.DataFrame) -> pd.DataFrame: stats = actor_director.groupby(
['actor_id', 'director_id'],
).agg(
count=('director_id', 'count'),
).reset_index()
return stats[stats['count'] >= 3][['actor_id', 'director_id']]
Результат:
Источник: https://leetcode.com/problems/actors-and-directors-who-cooperated-at-least-three-times/description/