value_counts, crosstab

Author: cmp0xff

pandas-stubs/__init__.pyi

@@ -69,6 +69,7 @@ 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,
@@ -80,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/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 (  # type: ignore[attr-defined] # isort: skip
-            value_counts,  # pyright: ignore[reportAttributeAccessIssue]
+            value_counts,  # pyright: ignore[reportAttributeAccessIssue,reportAttributeAccessIssue]
         )
 
-        return value_counts(self.to_numpy(), dropna=dropna)
+        return value_counts(
+            self.to_numpy(), dropna=dropna
+        )  # pyright: ignore[reportUnknownVariableType]
 
 
 DecimalArray._add_arithmetic_ops()