feat(pandas): add problem 620

Author: IndexSeek

README.md

@@ -30,7 +30,7 @@ Fiddling around with DataFusion, pandas, and PyArrow.
 
 | Problem # | Title | Difficulty | PyArrow | DataFusion | pandas |
 |----------:|:------|:-----------|:-------:|:----------:|:------:|
-| 620  | [Not Boring Movies](https://leetcode.com/problems/not-boring-movies) | Easy | ✅ | ✅ | ❌ |
+| 620  | [Not Boring Movies](https://leetcode.com/problems/not-boring-movies) | Easy | ✅ | ✅ | ✅ |
 | 1251 | [Average Selling Price](https://leetcode.com/problems/average-selling-price) | Easy | ✅ | ❌ | ❌ |
 | 1075 | [Project Employees I](https://leetcode.com/problems/project-employees-i) | Easy | ✅ | ❌ | ❌ |
 | 1633 | [Percentage of Users Attended a Contest](https://leetcode.com/problems/percentage-of-users-attended-a-contest) | Easy | ✅ | ❌ | ❌ |

problems/pandas.py

@@ -85,6 +85,25 @@ def problem_197(weather: pd.DataFrame) -> pd.DataFrame:
     return weather[mask][["id"]]
 
 
+def problem_620(cinema: pd.DataFrame) -> pd.DataFrame:
+    """Report movies with odd IDs and descriptions not equal to "boring".
+
+    Return the result table ordered by rating in descending order.
+
+    Parameters
+    ----------
+    cinema : pd.DataFrame
+        Table contains information about the name of a movie, genre, and its rating.
+
+    Returns
+    -------
+    pd.DataFrame
+
+    """
+    mask = (cinema["id"] % 2 == 1) & (cinema["description"] != "boring")
+    return cinema[mask].sort_values("rating", ascending=False)
+
+
 def problem_577(employee: pd.DataFrame, bonus: pd.DataFrame) -> pd.DataFrame:
     """Report the name and bonus amount of each employee with a bonus less than 1000.
 

tests/test_pandas.py

@@ -12,6 +12,7 @@
     problem_577,
     problem_584,
     problem_595,
+    problem_620,
     problem_1068,
     problem_1148,
     problem_1280,
@@ -357,6 +358,57 @@ def test_problem_595(input_data, expected_data):
         ), f"Expected table {expected_table}, but got {result}"
 
 
+@pytest.mark.parametrize(
+    "input_data, expected_data",
+    [
+        pytest.param(
+            {
+                "id": [1, 2, 3, 4],
+                "description": ["interesting", "boring", "exciting", "boring"],
+                "rating": [3, 1, 1, 1],
+            },
+            {
+                "id": [1, 3],
+                "description": ["interesting", "exciting"],
+                "rating": [3, 1],
+            },
+            id="happy_path_mixed_ids_and_descriptions",
+        ),
+        pytest.param(
+            {"id": [1, 3], "description": ["boring", "boring"], "rating": [1, 1]},
+            {"id": [], "description": [], "rating": []},
+            id="edge_case_all_boring",
+        ),
+        pytest.param(
+            {
+                "id": [2, 4],
+                "description": ["interesting", "exciting"],
+                "rating": [1, 1],
+            },
+            {"id": [], "description": [], "rating": []},
+            id="edge_case_no_odd_ids",
+        ),
+        pytest.param(
+            {"id": [1], "description": ["interesting"], "rating": [1]},
+            {"id": [1], "description": ["interesting"], "rating": [1]},
+            id="edge_case_single_row_matching",
+        ),
+        pytest.param(
+            {"id": [2], "description": ["boring"], "rating": [1]},
+            {"id": [], "description": [], "rating": []},
+            id="edge_case_single_row_not_matching",
+        ),
+    ],
+)
+def test_problem_620(input_data, expected_data):
+    table = pd.DataFrame(input_data)
+    expected_table = pd.DataFrame(expected_data)
+    result = problem_620(table).reset_index(drop=True)
+    assert_frame_equal(
+        result, expected_table, check_dtype=False, check_index_type=False
+    )
+
+
 @pytest.mark.parametrize(
     "input_data_1, input_data_2, expected_data",
     [