Completed Pandas5

01-DepartmentHighestSalary.py

@@ -0,0 +1,12 @@
+# Problem 1 - Department Highest Salary (https://leetcode.com/problems/department-highest-salary/ )
+import pandas as pd
+
+def department_highest_salary(employee: pd.DataFrame, department: pd.DataFrame) -> pd.DataFrame:
+    # Join Employee and Department Table via Inner Join
+    df = employee.merge(department, left_on = 'departmentId', right_on = 'id', how = 'inner')
+    # Compute Maximum Salary Per Department
+    max_salary = df.groupby('departmentId')['salary'].transform('max')
+    # Filter only those rows where salary is equal to max_salary
+    df = df[df['salary'] == max_salary]
+    # Return final dataframe with renamed columns
+    return df[['name_y', 'name_x', 'salary']].rename(columns = {'name_y' : 'Department', 'name_x' : 'Employee'})

02-RankScores.py

@@ -0,0 +1,8 @@
+# Problem 2 - Rank Scores ( https://leetcode.com/problems/rank-scores/ )
+import pandas as pd
+
+def order_scores(scores: pd.DataFrame) -> pd.DataFrame:
+    # Rank scores by dense rank method in pandas
+    scores['rank'] = scores['score'].rank(ascending = False, method = 'dense')
+    # Sort the table by rank in ascending order
+    return scores[['score', 'rank']].sort_values(by=['rank'])