16 lines
381 B
Python
16 lines
381 B
Python
|
import json
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
class NpEncoder(json.JSONEncoder):
|
||
|
|
def default(self, obj):
|
||
|
|
if isinstance(obj, np.integer):
|
||
|
|
return int(obj)
|
||
|
|
elif isinstance(obj, np.floating):
|
||
|
|
return float(obj)
|
||
|
|
elif isinstance(obj, np.ndarray):
|
||
|
|
return obj.tolist()
|
||
|
|
else:
|
||
|
|
return super(NpEncoder, self).default(obj)
|