How Many Veteran and Rookie Pilots Fit the Payroll

PandasHardSenior level~10 min

Problem

You are given a DataFrame `pilots` with columns `pilot_id`, `rank`, and `wage`. `pilot_id` uniquely identifies each pilot, `rank` is either `'Veteran'` or `'Rookie'`, and `wage` is the pilot's annual wage.

An airline has a hiring budget of `50000`. It first hires `Veteran` pilots, cheapest wage first (ties broken by smaller `pilot_id`), taking each as long as the running total of veteran wages stays within budget. With whatever budget remains, it then hires `Rookie` pilots under the same cheapest-first rule. Return two rows with columns `rank` and `hired_count`: the number of veterans hired and the number of rookies hired. Put the veteran row first.

Input data

Example rows — the live problem includes the full dataset.

pilots

pilot_id	rank	wage
1	Veteran	20000
2	Veteran	25000
3	Veteran	30000
4	Rookie	4000
5	Rookie	3000

Expected output

Your answer should return 2 rows with the columns rank, hired_count.

Starter code (Pandas (Python))

import pandas as pd

def how_many_veteran_and_rookie_pilots_fit_the_payroll(pilots) -> pd.DataFrame:
    # Your code here
    return pilots

Solve this Pandas question free

Write Pandas (Python) and run it instantly in your browser — even on your phone. No signup needed to try.

Solve it now → Create free account

Solution & explanation

Create a free account to unlock the optimal solution, a step-by-step explanation, and the hidden test cases that grade your answer.