diff --git a/Nth Highest Salary b/Nth Highest Salary new file mode 100644 index 0000000..3a2ce71 --- /dev/null +++ b/Nth Highest Salary @@ -0,0 +1,9 @@ +import pandas as pd + +def nth_highest_salary(employee: pd.DataFrame, N: int) -> pd.DataFrame: + + df = employee[['salary']].drop_duplicates() + if N > len(df) or N <= 0: + return pd.DataFrame({f'getNthHighestSalary({N})' : [None]}) + + return df.sort_values('salary', ascending = False).head(N).tail(1)[['salary']].rename(columns = {'salary': f'getNthHighestSalary({N})'}) \ No newline at end of file diff --git a/Second Highest Salary b/Second Highest Salary new file mode 100644 index 0000000..01ef62c --- /dev/null +++ b/Second Highest Salary @@ -0,0 +1,9 @@ +import pandas as pd + +def second_highest_salary(employee: pd.DataFrame) -> pd.DataFrame: + + df = employee[['salary']].drop_duplicates().sort_values(by = ['salary'], ascending=False) + if len(df) < 2: + return pd.DataFrame({'SecondHighestSalary': [None]}) + else: + return df.head(2).tail(1)[['salary']].rename(columns = {'salary' : 'SecondHighestSalary'}) \ No newline at end of file