Python json.dumps Numpy数组中的数据出错

最新推荐文章于 2023-08-18 10:07:20 发布

Replete

最新推荐文章于 2023-08-18 10:07:20 发布

阅读量584

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_38906715/article/details/126969726

版权

错误：TypeError: Object of type 'int64' is not JSON (或者float32)

在使用json格式保存数据时，经常会遇到xxx is not JSON serializable，也就是无法序列化某些对象格式，我所遇见的是我使用了numpy时，使用了np的数据格式，写入data后，json.dumps(data)失败，我们可以自己定定义对特定类型的对象的序列化，下面看下怎么定义和使用关于np数据类型的自定义。

1.首先，继承json.JSONEncoder，自定义序列化方法。

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)

2.使用dumps方法（我们可以直接把dict直接序列化为json对象）加上 cls=NpEncoder,data就可以正常序列化了

json.dumps(data, cls=NpEncoder)
其实，很简单，自定义一个序列化方法，然后dumps的时候加上cls=NpEncoder