ENH: Add dropna parameter to Series.unique() (fixes pandas-dev#61209)

Author: sahermuhamed1

pandas/core/base.py

@@ -16,6 +16,8 @@
 )
 
 import numpy as np
+from typing import Any
+from pandas._typing import ArrayLike
 
 from pandas._libs import lib
 from pandas._typing import (
@@ -1096,13 +1098,28 @@ def value_counts(
             dropna=dropna,
         )
 
-    def unique(self):
+    def unique(self, dropna: bool = True) -> ArrayLike:
+        """
+        Return unique values in the object.
+        
+        Parameters
+        ----------
+        dropna : bool, default True
+            If True, exclude NA/null values.
+            
+        Returns
+        -------
+        ndarray or ExtensionArray
+        """
         values = self._values
         if not isinstance(values, np.ndarray):
-            # i.e. ExtensionArray
+            # For ExtensionArray
             result = values.unique()
         else:
             result = algorithms.unique1d(values)
+        
+        if dropna:
+            result = result[~isna(result)]
         return result
 
     @final

pandas/core/series.py

@@ -2084,72 +2084,31 @@ def mode(self, dropna: bool = True) -> Series:
             dtype=self.dtype,
         ).__finalize__(self, method="mode")
 
-    def unique(self) -> ArrayLike:
+    def unique(self, dropna: bool = True) -> ArrayLike:
         """
         Return unique values of Series object.
-
-        Uniques are returned in order of appearance. Hash table-based unique,
-        therefore does NOT sort.
-
+        
+        Parameters
+        ----------
+        dropna : bool, default True
+            If True, exclude NA/null values.
+            
         Returns
         -------
         ndarray or ExtensionArray
-            The unique values returned as a NumPy array. See Notes.
-
-        See Also
-        --------
-        Series.drop_duplicates : Return Series with duplicate values removed.
-        unique : Top-level unique method for any 1-d array-like object.
-        Index.unique : Return Index with unique values from an Index object.
-
-        Notes
-        -----
-        Returns the unique values as a NumPy array. In case of an
-        extension-array backed Series, a new
-        :class:`~api.extensions.ExtensionArray` of that type with just
-        the unique values is returned. This includes
-
-            * Categorical
-            * Period
-            * Datetime with Timezone
-            * Datetime without Timezone
-            * Timedelta
-            * Interval
-            * Sparse
-            * IntegerNA
-
-        See Examples section.
-
+            The unique values returned as a NumPy array or ExtensionArray.
+            
         Examples
         --------
-        >>> pd.Series([2, 1, 3, 3], name="A").unique()
-        array([2, 1, 3])
-
-        >>> pd.Series([pd.Timestamp("2016-01-01") for _ in range(3)]).unique()
-        <DatetimeArray>
-        ['2016-01-01 00:00:00']
-        Length: 1, dtype: datetime64[s]
-
-        >>> pd.Series(
-        ...     [pd.Timestamp("2016-01-01", tz="US/Eastern") for _ in range(3)]
-        ... ).unique()
-        <DatetimeArray>
-        ['2016-01-01 00:00:00-05:00']
-        Length: 1, dtype: datetime64[s, US/Eastern]
-
-        An Categorical will return categories in the order of
-        appearance and with the same dtype.
-
-        >>> pd.Series(pd.Categorical(list("baabc"))).unique()
-        ['b', 'a', 'c']
-        Categories (3, object): ['a', 'b', 'c']
-        >>> pd.Series(
-        ...     pd.Categorical(list("baabc"), categories=list("abc"), ordered=True)
-        ... ).unique()
-        ['b', 'a', 'c']
-        Categories (3, object): ['a' < 'b' < 'c']
+        >>> s = pd.Series([1, 2, 2, pd.NA])
+        >>> s.unique()
+        array([1, 2])
+        
+        >>> s.unique(dropna=False)
+        array([1, 2, <NA>], dtype=object)
         """
-        return super().unique()
+        return super().unique(dropna=dropna)
+        
 
     @overload
     def drop_duplicates(