使用numpy库报错，TypeError: Object of type 'int64' is not JSON serializable.

项目中需要大量列表中的数字相加，于是使用numpy库，但是在json转换时报错。
开始以为是含有numpy类型的字典不能转换为json字符串，后来转换成list依然没有解决。
google后，找到解决办法，重新定义json类。

class NumpyEncoder(json.JSONEncoder):
    """ Special json encoder for numpy types """
    def default(self, obj):
        if isinstance(obj, (numpy.int_, numpy.intc, numpy.intp, numpy.int8,
                            numpy.int16, numpy.int32, numpy.int64, numpy.uint8,
                            numpy.uint16, numpy.uint32, numpy.uint64)):
            return int(obj)
        elif isinstance(obj, (numpy.float_, numpy.float16, numpy.float32,
                              numpy.float64)):
            return float(obj)
        elif isinstance(obj, (numpy.ndarray,)):
            return obj.tolist()
        return json.JSONEncoder.default(self, obj)

使用时:

import json
operation = json.dumps(operation, cls=NumpyEncoder)