Game Play Analysis II — LeetCode 512 Python Solution

The problem

For every player, report the device they used on the day of their first login. The Activity table has player_id (int), device_id (int), event_date (date) and games_played (int), with (player_id, event_date) as its primary key, so a player has at most one session per day. Return player_id and device_id.

Example

Input

Activity(player_id, device_id, event_date, games_played) = (1, 2, '2023-01-03', 5), (1, 5, '2023-01-06', 2), (2, 3, '2023-02-11', 1), (3, 1, '2023-01-05', 0), (3, 4, '2023-03-02', 4)

Output

(1, 2), (2, 3), (3, 1)

Explanation

Each row kept is the one whose event_date equals that player's earliest event_date.

Python solution

import pandas as pd

def game_play_analysis(activity: pd.DataFrame) -> pd.DataFrame:
    first = activity.groupby('player_id', as_index=False)['event_date'].min()
    res = activity.merge(first, on=['player_id', 'event_date'], how='inner')
    return res[['player_id', 'device_id']].drop_duplicates()

Complexity

Related problems