Skip to content

Commit c4d24b2

Browse files
committed
ty
1 parent 09553b0 commit c4d24b2

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

pandas-stubs/core/groupby/generic.pyi

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -359,3 +359,4 @@ class DataFrameGroupBy(GroupBy):
359359
ascending: bool = ...,
360360
dropna: bool = ...,
361361
) -> Series[float]: ...
362+
def __getattr__(self, name: str) -> Series: ...

tests/test_frame.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2097,3 +2097,14 @@ def test_groupby_and_transform() -> None:
20972097
check(assert_type(c2, pd.DataFrame), pd.DataFrame)
20982098
check(assert_type(c3, pd.Series), pd.Series)
20992099
check(assert_type(c4, pd.Series), pd.Series)
2100+
2101+
2102+
def test_getattr_and_dataframe_groupby() -> None:
2103+
df = pd.DataFrame(
2104+
{
2105+
'C': [1, 5, 5, 2, 5, 5],
2106+
'D': [2.0, 5.0, 8.0, 1.0, 2.0, 9.0],
2107+
}
2108+
)
2109+
gb = df.groupby('C').D
2110+
check(assert_type(gb, pd.Series), pd.Series)

0 commit comments

Comments
 (0)