Add BalancingLearner.to_dataframe

Author: basnijholt

adaptive/learner/balancing_learner.py

@@ -11,6 +11,14 @@
 from adaptive.notebook_integration import ensure_holoviews
 from adaptive.utils import cache_latest, named_product, restore
 
+try:
+    import pandas
+
+    with_pandas = True
+
+except ModuleNotFoundError:
+    with_pandas = False
+
 
 def dispatch(child_functions, arg):
     index, x = arg
@@ -381,6 +389,13 @@ def from_product(cls, f, learner_type, learner_kwargs, combos):
             learners.append(learner)
         return cls(learners, cdims=arguments)
 
+    def to_dataframe(self, kwargs):
+        if not with_pandas:
+            raise ImportError("pandas is not installed.")
+        dfs = [learner.to_dataframe(**kwargs) for learner in self.learners]
+        df = pandas.concat(dfs, axis=0, ignore_index=True)
+        return df
+
     def save(self, fname, compress=True):
         """Save the data of the child learners into pickle files
         in a directory.