Skip to content

Commit 76b7f32

Browse files
add code sample for pandas-dev#41999
1 parent 4ae8563 commit 76b7f32

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

bisect/41999.py

+31
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
# BUG?: 1.3 behavior change with groupby, apply and side effect #41999
2+
3+
import pandas as pd
4+
5+
print(pd.__version__)
6+
7+
8+
def apply_func(df):
9+
del df["MEMBER_ID"]
10+
11+
return df.head(1)
12+
13+
14+
df = pd.DataFrame.from_dict(
15+
{
16+
"MONTH": {0: "April", 1: "April", 2: "April"},
17+
"MEMBER_ID": {0: "Member A", 1: "Member A", 2: "Member B"},
18+
"ACTIVITY_CATEGORY": {
19+
0: "Activity 1",
20+
1: "Days off at homebase",
21+
2: "Activity 1",
22+
},
23+
"FTE": {0: 1.0, 1: 1.0, 2: 0.75},
24+
"FTE_OFF_DAYS": {0: 5, 1: 5, 2: 10},
25+
}
26+
)
27+
grp = df.groupby(["MONTH", "FTE"], observed=True)[
28+
["MEMBER_ID", "FTE_OFF_DAYS", "ACTIVITY_CATEGORY"]
29+
]
30+
result = grp.apply(apply_func)
31+
print(result)

0 commit comments

Comments
 (0)