diff --git a/NthHighestSalary.txt b/NthHighestSalary.txt new file mode 100644 index 0000000..61e16f9 --- /dev/null +++ b/NthHighestSalary.txt @@ -0,0 +1,8 @@ +import pandas as pd + +def nth_highest_salary(employee: pd.DataFrame, N: int) -> pd.DataFrame: + df = employee['salary'].drop_duplicates().sort_values(ascending=False) + if N > len(df) or N <= 0: + return pd.DataFrame({f'getNthHighestSalary({N})': [None]}) + df2 = df.iloc[N-1] + return pd.DataFrame({f'getNthHighestSalary({N})':[df2]}) \ No newline at end of file diff --git a/secondHighestSalary.txt b/secondHighestSalary.txt new file mode 100644 index 0000000..e6f074f --- /dev/null +++ b/secondHighestSalary.txt @@ -0,0 +1,7 @@ +import pandas as pd + +def second_highest_salary(employee: pd.DataFrame) -> pd.DataFrame: + df = employee['salary'].drop_duplicates().sort_values(ascending=False).head(2) + if len(df)<2: + return pd.DataFrame({'SecondHighestSalary':[None]}) + return pd.DataFrame(df).tail(1).rename(columns={'salary':'SecondHighestSalary'}) \ No newline at end of file