Backport PR #52330 on branch 2.0.x (ENH: Implement .dt.year for Arrow…

…ExtensionArray) (#52359) * Backport PR #52330: ENH: Implement .dt.year for ArrowExtensionArray * Update array.py --------- Co-authored-by: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Co-authored-by: Patrick Hoefler <61934744+phofl@users.noreply.github.com>
pandas-dev · Apr 2, 2023 · 9bf45db · 9bf45db
1 parent 7c56e05
commit 9bf45db
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pandas/core/arrays/arrow/array.py b/pandas/core/arrays/arrow/array.py
@@ -1945,6 +1945,10 @@ def _str_wrap(self, width, **kwargs):
             "str.wrap not supported with pd.ArrowDtype(pa.string())."
         )
 
+    @property
+    def _dt_year(self):
+        return type(self)(pc.year(self._data))
+
     @property
     def _dt_day(self):
         return type(self)(pc.day(self._data))

diff --git a/pandas/tests/extension/test_arrow.py b/pandas/tests/extension/test_arrow.py
@@ -2109,6 +2109,7 @@ def test_unsupported_dt(data):
 @pytest.mark.parametrize(
     "prop, expected",
     [
+        ["year", 2023],
         ["day", 2],
         ["day_of_week", 0],
         ["dayofweek", 0],