code sample for pandas-dev#38642

simonjayhawkins · Dec 23, 2020 · bfc2da2 · bfc2da2
1 parent c32b80b
commit bfc2da2
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/bisect/38642.py b/bisect/38642.py
@@ -0,0 +1,15 @@
+import pandas as pd
+
+print(pd.__version__)
+
+df = pd.DataFrame(
+    {"key": pd.Categorical(["b"] * 5, categories=["a", "b", "c", "d"]), "col": range(5)}
+)
+
+gb = df.groupby("key")
+
+result = list(gb.indices)
+print(result)
+# Out[12]: ['a', 'b', 'c', 'd']
+
+assert result == ["a", "b", "c", "d"]