From 6945eccd774932240f54f30b107176c8213ec87a Mon Sep 17 00:00:00 2001 From: kiran160195 <66009503+kiran160195@users.noreply.github.com> Date: Wed, 11 Jun 2025 16:33:16 -0500 Subject: [PATCH 1/2] Create problem1 --- problem1 | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 problem1 diff --git a/problem1 b/problem1 new file mode 100644 index 0000000..7aaaf5f --- /dev/null +++ b/problem1 @@ -0,0 +1,11 @@ +import pandas as pd + +def nth_highest_salary(employee: pd.DataFrame, N: int) -> pd.DataFrame: + employee = employee.drop_duplicates('salary') + employee['rank'] = employee['salary'].rank(method='dense', ascending=False) + result = employee[employee['rank'] == N][['salary']] + + if result.empty: + return pd.DataFrame({f'getNthHighestSalary({N})': [None]}) + else: + return pd.DataFrame({f'getNthHighestSalary({N})': [result.iloc[0]['salary']]}) From ea4371957cde8ef09f82efc0086e50932c85de0a Mon Sep 17 00:00:00 2001 From: kiran160195 <66009503+kiran160195@users.noreply.github.com> Date: Wed, 11 Jun 2025 18:14:38 -0500 Subject: [PATCH 2/2] Create problem2 --- problem2 | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 problem2 diff --git a/problem2 b/problem2 new file mode 100644 index 0000000..d37bb78 --- /dev/null +++ b/problem2 @@ -0,0 +1,12 @@ +import pandas as pd + +def second_highest_salary(employee: pd.DataFrame) -> pd.DataFrame: + secondSalary = employee.drop_duplicates('salary') + secondSalary['rank'] = secondSalary['salary'].rank(method='dense', ascending=False) + if len(secondSalary) >=2: + result = secondSalary[secondSalary['rank'] == 2][['salary']] + + else: + result = pd.DataFrame({'SecondHighestSalary': [None]}) + return result.rename(columns={'salary': 'SecondHighestSalary'}) +