Skip to content

Json encoder

NumpyEncoder

Bases: JSONEncoder

numpy encoder for json

Example

import json import numpy as np obj = {'a': np.array([1, 2, 3])} json.dumps(obj, cls=NumpyEncoder) '{"a": [1, 2, 3]}'

Source code in minto/v1/json_encoder.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class NumpyEncoder(json.JSONEncoder):
    """numpy encoder for json

    Example:
        >>> import json
        >>> import numpy as np
        >>> obj = {'a': np.array([1, 2, 3])}
        >>> json.dumps(obj, cls=NumpyEncoder)
        '{"a": [1, 2, 3]}'
    """

    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()
        return json.JSONEncoder.default(self, obj)