filtering

The DataFrame `watch_events` is preloaded. Write drama_events(df) that returns the rows where genre == 'drama', keeping only event_id, user, and minutes (in that order).

preloaded fixtures

import pandas as pd
import numpy as np

watch_events = pd.DataFrame({
    "event_id": range(1, 13),
    "user": ["alice", "bob", "alice", "carol", "bob", "alice", "carol", "dave", "bob", "carol", "dave", "alice"],
    "genre": ["drama", "comedy", "drama", "docs", "comedy", "thriller", "docs", "drama", "comedy", "thriller", "drama", "drama"],
    "watch_date": pd.to_datetime(["2024-01-05", "2024-01-06", "2024-01-10", "2024-01-12", "2024-01-15", "2024-01-20", "2024-02-01", "2024-02-03", "2024-02-10", "2024-02-15", "2024-02-20", "2024-02-28"]),
    "minutes": [45, 30, 60, 20, 25, 50, 15, 55, 40, 35, 60, 50],
})

Guest mode · solve history stays on this device; code isn’t saved. Sign in to save future work