2

我想扩展标准库 Python JSON 编码器以将复数列表编码为由 JSONLAB 正确解析的表示，JSONLAB是 MATLAB 的非官方 JSON 工具箱。简而言之，单个 Python 复数(x + yj)应编码为 JSON 复数对象，

{"_ArrayType_": "double",
 "_ArraySize_": [1, 1],
 "_ArrayIsComplex_": 1,
 "_ArrayData_": [x, y]}

并且复数列表也[(x1 + y1j), (x2+y2j), (x3+y3j)]应该成为单个 JSON 复数对象，

{"_ArrayType_": "double",
 "_ArraySize_": [1, 3],
 "_ArrayIsComplex_": 1,
 "_ArrayData_": [[x1, y1], [x2, y2], [x3, y3]]}

我已经能够正确编码一个复数，如下所示：

class ComplexEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, complex):
            return {"_ArrayType_": "double", "_ArraySize_": [1, 1],
                    "_ArrayIsComplex_": 1, "_ArrayData_": [obj.real, obj.imag]}
        return json.JSONEncoder.default(self, obj)

我尝试对复数列表的情况进行简单的扩展，但效果不佳：

class ComplexEncoder(json.JSONEncoder):
    def default(self, obj):
        # if obj is a list containing at least one complex number
        if isinstance(obj, list) and sum(map(lambda x: isinstance(x, complex),
                                             obj)):
            data = [[elem.real, elem.imag] for elem in obj]
            return {"_ArrayType_": "double", "_ArraySize_": [1, len(obj)],
                    "_ArrayIsComplex_": 1, "_ArrayData_": data}
        elif isinstance(obj, complex):
            return {"_ArrayType_": "double", "_ArraySize_": [1, 1],
                    "_ArrayIsComplex_": 1, "_ArrayData_": [obj.real, obj.imag]}
        return json.JSONEncoder.default(self, obj)

如果该类用于解析 Python 复数列表，则结果是 1x1 JSON 复数对象的 JSON 列表，而不是单个 JSON 复数对象。似乎 Python 对象是“自下而上”而不是“自上而下”解析的，复数在注意到它们在列表中之前转换为 1x1 JSON 复数对象。

如何最好地将复数列表编码为与 JSONLAB 兼容的复数对象？

4

1 回答 1

1

通过简要浏览simplejson源代码，除了自下而上之外，似乎没有其他简单的方法可以以任何方式对 JSON 进行编码。我建议两种可能的解决方案：

  • 使用不可迭代或非列表对象（即json模块不解析的对象）来包含复数列表，以便将其编码为default()and not json.JSONEncoder。
  • 在编码其余部分之前对复数列表进行编码，这样 json 编码器就不必在子对象之前处理对父对象的编码。
于 2013-10-16T20:08:31.820 回答