feat(pandas): add problem 1148

Author: IndexSeek

README.md

@@ -9,7 +9,7 @@ Fiddling around with DataFusion, pandas, and PyArrow.
 | 1757      | [Recyclable and Low Fat Products](https://leetcode.com/problems/recyclable-and-low-fat-products) | Easy | ✅ | ❌ | ✅ |
 | 584       | [Find Customer Referee](https://leetcode.com/problems/find-customer-referee) | Easy | ✅ | ❌ | ✅ |
 | 595       | [Big Countries](https://leetcode.com/problems/big-countries) | Easy | ✅ | ❌ | ✅ |
-| 1148      | [Article Views I](https://leetcode.com/problems/article-views-i) | Easy | ✅ | ❌ | ❌ |
+| 1148      | [Article Views I](https://leetcode.com/problems/article-views-i) | Easy | ✅ | ❌ | ✅ |
 | 1683      | [Invalid Tweets](https://leetcode.com/problems/invalid-tweets) | Easy | ✅ | ❌ | ❌ |
 
 ## Basic Joins

problems/pandas.py

@@ -75,6 +75,29 @@ def problem_595(world: pd.DataFrame) -> pd.DataFrame:
     return world[columns]
 
 
+def problem_1148(views: pd.DataFrame) -> pd.DataFrame:
+    """Find all the authors that viewed at least one of their own articles.
+
+    Return the result table sorted by id in ascending order.
+
+    Parameters
+    ----------
+    views : pd.DataFrame
+        Table logs viewers viewing articles by authors on specific dates.
+
+    Returns
+    -------
+    pd.DataFrame
+
+    """
+    return (
+        views[views["author_id"] == views["viewer_id"]][["author_id"]]
+        .drop_duplicates()
+        .sort_values("author_id")
+        .rename(columns={"author_id": "id"})
+    )
+
+
 def problem_1321(customer: pd.DataFrame) -> pd.DataFrame:
     """Compute the moving average of how much the customer paid in a seven days window.
 

tests/test_pandas.py

@@ -7,6 +7,7 @@
     problem_176,
     problem_584,
     problem_595,
+    problem_1148,
     problem_1321,
     problem_1757,
 )
@@ -142,6 +143,46 @@ 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(
+            {
+                "article_id": [1, 2, 3],
+                "author_id": [3, 7, 4],
+                "viewer_id": [3, 7, 4],
+                "view_date": [
+                    datetime(2019, 8, 1),
+                    datetime(2019, 8, 1),
+                    datetime(2019, 7, 21),
+                ],
+            },
+            {
+                "id": [3, 4, 7],
+            },
+            id="happy_path",
+        ),
+        pytest.param(
+            {
+                "article_id": [1, 2],
+                "author_id": [3, 7],
+                "viewer_id": [3, 7],
+                "view_date": [datetime(2019, 8, 1), datetime(2019, 8, 1)],
+            },
+            {
+                "id": [3, 7],
+            },
+            id="all_match",
+        ),
+    ],
+)
+def test_problem_1148(input_data, expected_data):
+    table = pd.DataFrame(input_data)
+    expected_table = pd.DataFrame(expected_data)
+    result = problem_1148(table).reset_index(drop=True)
+    assert result.equals(expected_table)
+
+
 @pytest.mark.parametrize(
     "input_data, expected_data",
     [