feat:GridTableForm can be invoked on datasets.

CNS1mple · Dec 10, 2021 · 6371c93 · 6371c93
1 parent e4e1d05
commit 6371c93
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/MathematicaForPredictionUtilities.m b/MathematicaForPredictionUtilities.m
@@ -446,6 +446,18 @@ Mathematica is (C) Copyright 1988-2016 Wolfram Research, Inc.
       ]
     ];
 
+GridTableForm[ds_Dataset] :=
+    Which[
+      AssociationQ[Normal[ds]] && AssociationQ[Normal[ds][[1]]],
+      GridTableForm[Normal@ds[Values, Values], TableHeadings -> Normal[Keys[ds[[1]]]]],
+
+      ListQ[Normal[ds]] && AssociationQ[Normal[ds][[1]]],
+      GridTableForm[Normal@ds[Values], TableHeadings -> Normal[Keys[ds[[1]]]]],
+
+      True,
+      GridTableForm[Normal@ds]
+    ];
+
 GridTableForm[___] :=
     Block[{},
       Message[GridTableForm::nargs];