balance = {} def withdraw(user_id, amount): if balance.get(user_id, 0) >= amount: balance[user_id] = balance.get(user_id, 0) - amount return True return False