Skip to content

Commit 14d2735

Browse files
code sample for pandas-dev#37641
1 parent d1a635d commit 14d2735

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

bisect/37641.py

+20
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
import pandas as pd
2+
3+
print(pd.__version__)
4+
5+
6+
arrays = [["val1", "val1", "val2"], ["val1", "val1", "val2"]]
7+
index = pd.MultiIndex.from_arrays(arrays, names=("idx1", "idx2"))
8+
9+
s = pd.Series([1, 2, 3], index=index)
10+
11+
res = s.groupby(["idx1", "idx2"], group_keys=False).rolling(1).mean()
12+
print(res)
13+
14+
mi = pd.MultiIndex.from_tuples(
15+
[("val1", "val1"), ("val1", "val1"), ("val2", "val2")], names=["idx1", "idx2"]
16+
)
17+
18+
import pandas.testing as tm
19+
20+
tm.assert_index_equal(res.index, mi)

0 commit comments

Comments
 (0)