Issue 21932: Added Regression Test for index of pivot_table on empty …

…table (pandas-dev#52809)
Daquisu · Jul 8, 2023 · bf535f1 · bf535f1
1 parent 92249ab
commit bf535f1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pandas/tests/reshape/test_pivot.py b/pandas/tests/reshape/test_pivot.py
@@ -2598,3 +2598,11 @@ def test_pivot_not_changing_index_name(self):
         expected = df.copy(deep=True)
         df.pivot(index="one", columns="two", values="three")
         tm.assert_frame_equal(df, expected)
+
+    def test_pivot_table_empty_dataframe_correct_index(self):
+        # GH 21932
+        df = DataFrame([], columns=["a", "b", "value"])
+        pivot = df.pivot_table(index="a", columns="b", values="value", aggfunc="count")
+
+        expected = Index([], dtype="object", name="b")
+        tm.assert_index_equal(pivot.columns, expected)