Merge pull request #58 from pnnl/Fix_boolean_value_plot

Fix boolean value plotting
pnnl · Nov 8, 2024 · 92650df · 92650df
2 parents 34acb9b + 6c64ca5
commit 92650df
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/constrain/checklib.py b/constrain/checklib.py
@@ -213,6 +213,9 @@ def all_plot_obo(self, plt_pts, fig_size):
                 if pt_nan[pt]:
                     self.df[pt].plot(ax=axx, marker=".")
                 else:
+                    # check if values in the series are boolean
+                    if self.df[pt].apply(lambda x: isinstance(x, bool)).all():
+                        self.df[pt] = self.df[pt].astype(int)
                     self.df[pt].plot(ax=axx)
                 plt.title(f"All samples - {pt} - {self.__class__.__name__}")
                 i += 1
@@ -326,6 +329,9 @@ def day_plot_obo(self, plt_pts, fig_size):
                 if pt_nan[pt]:
                     plotdaydf[pt].plot(ax=axx, marker=".")
                 else:
+                    # check if values in the series are boolean
+                    if self.df[pt].apply(lambda x: isinstance(x, bool)).all():
+                        self.df[pt] = self.df[pt].astype(int)
                     plotdaydf[pt].plot(ax=axx)
                 plt.title(f"Example day - {pt} - {self.__class__.__name__}")
                 i += 1