modified: pandas/tests/reshape/test_pivot.py

	- Added test :callable:`test_pivot_table_index_and_column_keys_with_nan` to test that null index/column keys are kept when `dropna=False`, and removed when `dropna=True`.

Author: it176131

pandas/tests/reshape/test_pivot.py

@@ -2529,6 +2529,63 @@ def test_pivot_table_aggfunc_nunique_with_different_values(self):
 
         tm.assert_frame_equal(result, expected)
 
+    def test_pivot_table_index_and_column_keys_with_nan(self, dropna: bool) -> None:
+        """Index/column keys containing NA values depend on ``dropna``.
+
+        Input data
+        ----------
+            row  col  val
+        0  NaN  0.0    0
+        1  0.0  1.0    1
+        2  1.0  2.0    2
+        3  2.0  3.0    3
+        4  3.0  NaN    4
+
+        ``dropna=True``: Index/column keys containing NA values will be dropped.
+        Expected output
+        ---------------
+        col  1.0  2.0  3.0
+        row
+        0.0  1.0  NaN  NaN
+        1.0  NaN  2.0  NaN
+        2.0  NaN  NaN  3.0
+
+        ``dropna=False``: Index/columns should exist if any non-null values.
+        Expected output
+        ---------------
+        col  0.0  1.0  2.0  3.0  NaN
+        row
+        0.0  NaN  1.0  NaN  NaN  NaN
+        1.0  NaN  NaN  2.0  NaN  NaN
+        2.0  NaN  NaN  NaN  3.0  NaN
+        3.0  NaN  NaN  NaN  NaN  4.0
+        NaN  0.0  NaN  NaN  NaN  NaN
+        """
+        data = {"row": [None, *range(4)], "col": [*range(4), None], "val": range(5)}
+        df = DataFrame(data)
+        actual = df.pivot_table(values="val", index="row", columns="col", dropna=dropna)
+        e_index = [None, *range(4)]
+        e_columns = [*range(4), None]
+        e_data = np.zeros(shape=(5, 5))
+        e_data.fill(np.NaN)
+        np.fill_diagonal(a=e_data, val=range(5))
+        expected = DataFrame(
+            data=e_data,
+            index=Index(data=e_index, name="row"),
+            columns=Index(data=e_columns, name="col"),
+        ).sort_index()
+        if dropna:
+            expected = (
+                # Drop null index/column keys.
+                expected.loc[expected.index.notna(), expected.columns.notna()]
+                # Drop null rows.
+                .dropna(axis="index", how="all")
+                # Drop null columns.
+                .dropna(axis="columns", how="all")
+            )
+
+        tm.assert_frame_equal(left=actual, right=expected)
+
 
 class TestPivot:
     def test_pivot(self):