🎨 implement diff within extensionarray

Author: Marco Gorelli

doc/source/whatsnew/v1.0.0.rst

@@ -1146,7 +1146,7 @@ ExtensionArray
 - Bug in :class:`arrays.PandasArray` when setting a scalar string (:issue:`28118`, :issue:`28150`).
 - Bug where nullable integers could not be compared to strings (:issue:`28930`)
 - Bug where :class:`DataFrame` constructor raised ``ValueError`` with list-like data and ``dtype`` specified (:issue:`30280`)
-- Bug in :meth:`Series.diff` was always setting the ``dtype`` to ``Object`` (:issue:`30889`)
+- Bug in :meth:`Series.diff` was always setting the ``dtype`` to ``object`` (:issue:`30889`)
 
 
 Other

pandas/core/arrays/base.py

@@ -24,7 +24,7 @@
 from pandas.core.dtypes.missing import isna
 
 from pandas.core import ops
-from pandas.core.algorithms import _factorize_array, unique
+from pandas.core.algorithms import _factorize_array, diff, unique
 from pandas.core.missing import backfill_1d, pad_1d
 from pandas.core.sorting import nargsort
 
@@ -516,6 +516,27 @@ def argsort(
         result = nargsort(self, kind=kind, ascending=ascending, na_position="last")
         return result
 
+    def diff(self, periods: int = 1):
+        """
+        First discrete difference of element.
+
+        Calculates the difference of a ExtensionArray element compared with another
+        element in the ExtensionArray (default is element in previous row).
+
+        Parameters
+        ----------
+        periods : int, default 1
+            Periods to shift for calculating difference, accepts negative
+            values.
+
+        Returns
+        -------
+        ExtensionArray
+            First differences of the ExtensionArray.
+        """
+
+        return self._from_sequence(diff(self, periods), dtype=self.dtype)
+
     def fillna(self, value=None, method=None, limit=None):
         """
         Fill NA/NaN values using the specified method.

pandas/core/series.py

@@ -2316,13 +2316,11 @@ def diff(self, periods=1) -> "Series":
         5    NaN
         dtype: float64
         """
-        result = algorithms.diff(com.values_from_object(self), periods)
         if is_extension_array_dtype(self.dtype) and not is_datetime64tz_dtype(
             self.dtype
         ):
-            return self._constructor(
-                result, index=self.index, dtype=self.dtype
-            ).__finalize__(self)
+            return self.values.diff(periods=periods)
+        result = algorithms.diff(com.values_from_object(self), periods)
         return self._constructor(result, index=self.index).__finalize__(self)
 
     def autocorr(self, lag=1) -> float: