diff --git a/Nth_Highest_Salary.py b/Nth_Highest_Salary.py new file mode 100644 index 0000000..cbbebc2 --- /dev/null +++ b/Nth_Highest_Salary.py @@ -0,0 +1,17 @@ +import pandas as pd + +def nth_highest_salary(employee: pd.DataFrame, N: int) -> pd.DataFrame: + result_set = set() + for i in range(len(employee)): + salary = employee['salary'][i] + result_set.add(salary) + result =[] + for element in result_set: + result.append(element) + + result.sort(reverse = True) + #invalid + if N > len(result) or N <= 0: + return pd.DataFrame({f'getNthHighestSalary({N})' : [None]}) + + return pd.DataFrame({f'getNthHighestSalary({N})' : [result[N-1]]}) \ No newline at end of file diff --git a/Second_Highest_Salary.py b/Second_Highest_Salary.py new file mode 100644 index 0000000..df822b9 --- /dev/null +++ b/Second_Highest_Salary.py @@ -0,0 +1,18 @@ +import pandas as pd + +def second_highest_salary(employee: pd.DataFrame) -> pd.DataFrame: + all_sal = set() + for i in range(len(employee)): + sal = employee['salary'][i] + all_sal.add(sal) + + result =[] + for salary in all_sal: + result.append(salary) + + result.sort(reverse = True) + if len(result) < 2: + return pd.DataFrame([None] , columns =['SecondHighestSalary']) + return pd.DataFrame([result[1]] , columns = ['SecondHighestSalary']) + + \ No newline at end of file