value_counts, crosstab

Author: cmp0xff

pandas-stubs/__init__.pyi

@@ -65,13 +65,11 @@ from pandas.core.api import (
     to_timedelta as to_timedelta,
     unique as unique,
 )
-from pandas.core.api import (
-    value_counts as value_counts,  # pyright: ignore[reportUnknownVariableType]
-)
 from pandas.core.arrays.sparse import SparseDtype as SparseDtype
 from pandas.core.computation.api import eval as eval
 from pandas.core.reshape.api import (
     concat as concat,
+    crosstab as crosstab,
     from_dummies as from_dummies,
     get_dummies as get_dummies,
     lreshape as lreshape,
@@ -83,9 +81,6 @@ from pandas.core.reshape.api import (
     qcut as qcut,
     wide_to_long as wide_to_long,
 )
-from pandas.core.reshape.api import (
-    crosstab as crosstab,  # pyright: ignore[reportUnknownVariableType]
-)
 from pandas.core.reshape.api import (
     cut as cut,  # pyright: ignore[reportUnknownVariableType]
 )

pandas-stubs/core/algorithms.pyi

@@ -19,7 +19,6 @@ from pandas.core.series import Series
 
 from pandas._typing import (
     T_EXTENSION_ARRAY,
-    AnyArrayLike,
     GenericT,
     IntervalT,
     TakeIndexer,
@@ -88,14 +87,6 @@ def factorize(
     use_na_sentinel: bool = ...,
     size_hint: int | None = ...,
 ) -> tuple[np_1darray_int64, Categorical]: ...
-def value_counts(
-    values: AnyArrayLike | list | tuple,
-    sort: bool = True,
-    ascending: bool = False,
-    normalize: bool = False,
-    bins: int | None = None,
-    dropna: bool = True,
-) -> Series: ...
 def take(
     arr: np_ndarray[Any] | ExtensionArray | Index | Series,
     indices: TakeIndexer,

pandas-stubs/core/api.pyi

@@ -4,9 +4,6 @@ from pandas.core.algorithms import (
     factorize as factorize,
     unique as unique,
 )
-from pandas.core.algorithms import (
-    value_counts as value_counts,  # pyright: ignore[reportUnknownVariableType]
-)
 from pandas.core.arrays import Categorical as Categorical
 from pandas.core.arrays.arrow.dtype import ArrowDtype as ArrowDtype
 from pandas.core.arrays.boolean import BooleanDtype as BooleanDtype

pandas-stubs/core/reshape/pivot.pyi

@@ -28,6 +28,7 @@ from pandas._typing import (
     Label,
     Scalar,
     ScalarT,
+    SequenceNotStr,
     np_ndarray,
 )
 
@@ -121,11 +122,14 @@ def pivot(
     columns: _NonIterableHashable | Sequence[HashableT2] = ...,
     values: _NonIterableHashable | Sequence[HashableT3] = ...,
 ) -> DataFrame: ...
+
+_Values: TypeAlias = SequenceNotStr[Any] | _ExtendedAnyArrayLike
+
 @overload
 def crosstab(
-    index: list | _ExtendedAnyArrayLike | list[Sequence | _ExtendedAnyArrayLike],
-    columns: list | _ExtendedAnyArrayLike | list[Sequence | _ExtendedAnyArrayLike],
-    values: list | _ExtendedAnyArrayLike,
+    index: _Values | list[_Values],
+    columns: _Values | list[_Values],
+    values: _Values,
     rownames: list[HashableT1] | None = ...,
     colnames: list[HashableT2] | None = ...,
     *,
@@ -137,8 +141,8 @@ def crosstab(
 ) -> DataFrame: ...
 @overload
 def crosstab(
-    index: list | _ExtendedAnyArrayLike | list[Sequence | _ExtendedAnyArrayLike],
-    columns: list | _ExtendedAnyArrayLike | list[Sequence | _ExtendedAnyArrayLike],
+    index: _Values | list[_Values],
+    columns: _Values | list[_Values],
     values: None = None,
     rownames: list[HashableT1] | None = ...,
     colnames: list[HashableT2] | None = ...,

tests/extension/decimal/array.py

@@ -345,10 +345,12 @@ def convert_values(param: Any) -> ExtensionArray | list[Any]:
 
     def value_counts(self, dropna: bool = True) -> Series:
         from pandas.core.algorithms import (
-            value_counts,  # pyright: ignore[reportUnknownVariableType]
+            value_counts,  # pyright: ignore[reportAttributeAccessIssue,reportUnknownVariableType]
         )
 
-        return value_counts(self.to_numpy(), dropna=dropna)
+        return value_counts(
+            self.to_numpy(), dropna=dropna
+        )  # pyright: ignore[reportUnknownVariableType]
 
 
 DecimalArray._add_arithmetic_ops()