Merge branch 'bar_plot_stacking_fix' into subplot_label_count_fix

Author: eicchen

pandas/plotting/_matplotlib/core.py

@@ -1943,13 +1943,14 @@ def _make_plot(self, fig: Figure) -> None:
 
         self.subplots: list[Any]
 
-        if bool(self.subplots) and self.stacked:
-            for i, sub_plot in enumerate(self.subplots):
-                if len(sub_plot) <= 1:
-                    continue
-                for plot in sub_plot:
-                    _stacked_subplots_ind[int(plot)] = i
-                _stacked_subplots_offsets.append([0, 0])
+        if type(self.subplots) != bool:
+            if bool(self.subplots) and self.stacked:
+                for i, sub_plot in enumerate(self.subplots):
+                    if len(sub_plot) <= 1:
+                        continue
+                    for plot in sub_plot:
+                        _stacked_subplots_ind[int(plot)] = i
+                    _stacked_subplots_offsets.append([0, 0])
 
         for i, (label, y) in enumerate(self._iter_data(data=data)):
             ax = self._get_ax(i)

pandas/tests/plotting/test_misc.py

@@ -727,7 +727,11 @@ def _df_bar_subplot_checker(df_bar_data, df_bar_df, subplot_data_df, subplot_col
         ].reset_index()
         for i in range(len(subplot_columns))
     ]
-    expected_total_height = df_bar_df.loc[:, subplot_columns].sum(axis=1)
+
+    if len(subplot_columns) == 1:
+        expected_total_height = df_bar_df.loc[:, subplot_columns[0]]
+    else:
+        expected_total_height = df_bar_df.loc[:, subplot_columns].sum(axis=1)
 
     for i in range(len(subplot_columns)):
         sliced_df = subplot_sliced_by_source[i]
@@ -743,7 +747,6 @@ def _df_bar_subplot_checker(df_bar_data, df_bar_df, subplot_data_df, subplot_col
             tm.assert_series_equal(
                 height_iter, expected_total_height, check_names=False, check_dtype=False
             )
-
         else:
             # Checks each preceding bar ends where the next one starts
             next_start_coord = subplot_sliced_by_source[i + 1]["y_coord"]
@@ -818,6 +821,18 @@ def test_bar_2_subplots_1_triple_stacked(df_bar_data, df_bar_df, subplot_divisio
         )
 
 
+def test_bar_subplots_stacking_bool(df_bar_data, df_bar_df):
+    subplot_division = [("A"), ("B"), ("C"), ("D")]
+    ax = df_bar_df.plot(subplots=True, kind="bar", stacked=True)
+    subplot_data_df_list = _df_bar_xyheight_from_ax_helper(
+        df_bar_data, ax, subplot_division
+    )
+    for i in range(len(subplot_data_df_list)):
+        _df_bar_subplot_checker(
+            df_bar_data, df_bar_df, subplot_data_df_list[i], subplot_division[i]
+        )
+
+
 def test_plot_bar_label_count_default():
     df = DataFrame(
         [(30, 10, 10, 10), (20, 20, 20, 20), (10, 30, 30, 10)], columns=list("ABCD")