From d65cfbf76d3a87aee7bb5c67d284ba5e04dcad79 Mon Sep 17 00:00:00 2001
From: jay <jlaura@asu.edu>
Date: Tue, 3 Jul 2018 06:43:23 -0700
Subject: [PATCH] Adds the np encoder for JSON arrays
---
plio/io/io_json.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/plio/io/io_json.py b/plio/io/io_json.py
index a463e6c..161a68e 100644
--- a/plio/io/io_json.py
+++ b/plio/io/io_json.py
@@ -1,6 +1,15 @@
import json
+import numpy as np
+class NumpyEncoder(json.JSONEncoder):
+ def default(self, obj):
+ if isinstance(obj, np.ndarray):
+ return obj.tolist()
+ elif isinstance(obj, datetime.date):
+ return obj.isoformat()
+ return json.JSONEncoder.default(self, obj)
+
def read_json(inputfile):
"""
Read the input json file into a python dictionary.
--
GitLab