Use df.to_json(orient='table'

Author: gnestor

jupyterlab_table/__init__.py

@@ -89,12 +89,7 @@ def schema(self, schema):
     def _ipython_display_(self):
         bundle = {
             'application/vnd.dataresource+json': {
-                'resources': [
-                    {
-                        'schema': self.schema,
-                        'data': prepare_data(self.data)
-                    }
-                ]
+                'resources': [prepare_data(self.data, self.schema)]
             },
             'text/plain': '<jupyterlab_table.JSONTable object>'
         }

jupyterlab_table/utils.py

@@ -2,6 +2,7 @@
 import codecs
 import collections
 import os.path
+import json
 import pandas as pd
 
 
@@ -72,10 +73,14 @@ def sanitize_dataframe(df):
     return df
 
 
-def prepare_data(data=None):
-    """Prepare Plotly data from Pandas DataFrame."""
+def prepare_data(data=None, schema=None):
+    """Prepare JSONTable data from Pandas DataFrame."""
 
-    if isinstance(data, list):
-        return data
-    data = sanitize_dataframe(data)
-    return data.to_dict(orient='records')
+    if isinstance(data, pd.DataFrame):
+        data = sanitize_dataframe(data)
+        data = data.to_json(orient='table')
+        return json.loads(data)
+    return {
+        'data': data,
+        'schema': schema
+    }