Fixed crash when "subplots=True" is used

Author: eicchen

pandas/plotting/_matplotlib/core.py

@@ -1934,13 +1934,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]
+    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]
@@ -816,3 +820,15 @@ def test_bar_2_subplots_1_triple_stacked(df_bar_data, df_bar_df, subplot_divisio
         _df_bar_subplot_checker(
             df_bar_data, df_bar_df, subplot_data_df_list[i], subplot_division[i]
         )
+
+
+def test_bar_subplots_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]
+        )