Merge двух DataFrame + фильтрация

EasyML coding
05:00
Лучше работает на десктопе
Pandasmergegroupbyфильтрация

Даны два набора данных: пользователи и заказы (как словари для конструктора DataFrame).

Объедините их по user_id и верните имена пользователей, у которых сумма заказов больше заданного порога.

Задача основана на вопросах с собеседований в МТС и Lamoda.

Сигнатура

def filter_by_total(users: dict, orders: dict, threshold: int) -> list[str]:

Примеры

Пример 1

Вход:
users = {"user_id":[1,2,3],"name":["Alice","Bob","Charlie"]}
orders = {"user_id":[1,1,2,3,3],"amount":[100,200,150,50,60]}
threshold = 250
Выход:["Alice"]

Пример 2

Вход:
users = {"user_id":[1,2,3],"name":["Alice","Bob","Charlie"]}
orders = {"user_id":[1,1,2,3,3],"amount":[100,200,150,50,60]}
threshold = 100
Выход:["Alice","Bob","Charlie"]

Пример 3

Вход:
users = {"user_id":[1,2],"name":["Alice","Bob"]}
orders = {"user_id":[1,2],"amount":[500,500]}
threshold = 499
Выход:["Alice","Bob"]
Консоль
Нажмите Run или Ctrl+Enter для запуска