diff --git a/pandas-stubs/core/groupby/generic.pyi b/pandas-stubs/core/groupby/generic.pyi index d2d025de0..1cfab095e 100644 --- a/pandas-stubs/core/groupby/generic.pyi +++ b/pandas-stubs/core/groupby/generic.pyi @@ -272,7 +272,7 @@ class DataFrameGroupBy(GroupBy[DataFrame], Generic[ByT, _TT]): @overload def __getitem__( # pyright: ignore[reportIncompatibleMethodOverride] self, key: Iterable[Hashable] - ) -> DataFrameGroupBy[ByT, bool]: ... + ) -> DataFrameGroupBy[ByT, _TT]: ... def nunique(self, dropna: bool = ...) -> DataFrame: ... def idxmax( self, diff --git a/tests/test_groupby.py b/tests/test_groupby.py index cc81feb07..e6dbd0e12 100644 --- a/tests/test_groupby.py +++ b/tests/test_groupby.py @@ -1093,3 +1093,12 @@ def test_series_value_counts() -> None: Series, np.float64, ) + + +def test_dataframe_value_counts() -> None: + df = DataFrame({"a": [1, 1, 2], "b": [4, 5, 6], "c": [5, 5, 2]}) + check( + assert_type(df.groupby("a")[["b", "c"]].value_counts(), "Series[int]"), + Series, + np.int64, + )