Backport PR pandas-dev#52330: ENH: Implement .dt.year for ArrowExtens…

…ionArray
meeseeksmachine · Apr 2, 2023 · ee2c84f · ee2c84f
1 parent c922324
commit ee2c84f
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._pa_array))
+
     @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],