feat(pandas): add problem 610

Author: IndexSeek

README.md

@@ -48,7 +48,7 @@ Fiddling around with DataFusion, pandas, and PyArrow.
 |-------------:|:---------------------------------------------------------------------------------------------------------------------------------------------|:-------------|:-------------|:---------|:----------|
 |         1731 | [The Number of Employees Which Report to Each Employee](https://leetcode.com/problems/the-number-of-employees-which-report-to-each-employee) | Easy         | ❌           | ❌       | ✅        |
 |         1789 | [Primary Department for Each Employee](https://leetcode.com/problems/primary-department-for-each-employee)                                   | Easy         | ❌           | ❌       | ✅        |
-|          610 | [Triangle Judgement](https://leetcode.com/problems/triangle-judgement)                                                                       | Easy         | ❌           | ❌       | ✅        |
+|          610 | [Triangle Judgement](https://leetcode.com/problems/triangle-judgement)                                                                       | Easy         | ❌           | ✅       | ✅        |
 |          180 | [Consecutive Numbers](https://leetcode.com/problems/consecutive-numbers)                                                                     | Medium       | ✅           | ✅       | ✅        |
 |         1164 | [Product Price at a Given Date](https://leetcode.com/problems/product-price-at-a-given-date)                                                 | Medium       | ❌           | ❌       | ✅        |
 |         1204 | [Last Person to Fit in the Bus](https://leetcode.com/problems/last-person-to-fit-in-the-bus)                                                 | Medium       | ❌           | ❌       | ✅        |

problems/pandas.py

@@ -258,6 +258,30 @@ def problem_596(courses: pd.DataFrame) -> pd.DataFrame:
     )
 
 
+def problem_610(triangle: pd.DataFrame) -> pd.DataFrame:
+    """Report for every three line segments whether they can form a triangle.
+
+    Return the result table in any order.
+
+    Parameters
+    ----------
+    triangle : pd.DataFrame
+        Table contains the lengths of three line segments (x, y, z).
+
+    Returns
+    -------
+    pd.DataFrame
+
+    """
+    triangle["triangle"] = (
+        (triangle["x"] + triangle["y"] > triangle["z"])
+        & (triangle["x"] + triangle["z"] > triangle["y"])
+        & (triangle["y"] + triangle["z"] > triangle["x"])
+    )
+    triangle["triangle"] = triangle["triangle"].map({True: "Yes", False: "No"})
+    return triangle
+
+
 def problem_1068(sales: pd.DataFrame, product: pd.DataFrame) -> pd.DataFrame:
     """Report the product_name, year, and price for each sale_id in the Sales table.
 

tests/test_pandas.py

@@ -499,6 +499,50 @@ def test_problem_596(input_data, expected_data):
     )
 
 
+@pytest.mark.parametrize(
+    "input_data, expected_data",
+    [
+        pytest.param(
+            {"x": [3], "y": [4], "z": [5]},
+            {"x": [3], "y": [4], "z": [5], "triangle": ["Yes"]},
+            id="valid_triangle",
+        ),
+        pytest.param(
+            {"x": [5], "y": [5], "z": [5]},
+            {"x": [5], "y": [5], "z": [5], "triangle": ["Yes"]},
+            id="equilateral_triangle",
+        ),
+        pytest.param(
+            {"x": [2], "y": [2], "z": [3]},
+            {"x": [2], "y": [2], "z": [3], "triangle": ["Yes"]},
+            id="isosceles_triangle",
+        ),
+        pytest.param(
+            {"x": [1], "y": [1], "z": [2]},
+            {"x": [1], "y": [1], "z": [2], "triangle": ["No"]},
+            id="degenerate_triangle",
+        ),
+        pytest.param(
+            {"x": [0], "y": [0], "z": [0]},
+            {"x": [0], "y": [0], "z": [0], "triangle": ["No"]},
+            id="zero_length_sides",
+        ),
+        pytest.param(
+            {"x": [1], "y": [2], "z": [3]},
+            {"x": [1], "y": [2], "z": [3], "triangle": ["No"]},
+            id="non_triangle",
+        ),
+    ],
+)
+def test_problem_610(input_data, expected_data):
+    table = pd.DataFrame(input_data)
+    expected_table = pd.DataFrame(expected_data)
+    result = problem_610(table).reset_index(drop=True)
+    assert_frame_equal(
+        result, expected_table, check_dtype=False, check_index_type=False
+    )
+
+
 @pytest.mark.parametrize(
     "input_data, expected_data",
     [