如何使用json.dumps()在字典中获取排序列表

Question

我有以下问题:有一个像这样的python字典:

{"qqq": [{"bbb": "111"}, {"aaa": "333"}], "zzz": {"bbb": [5, 2, 1, 9]}}

我想获得一个有序的json对象，如:

'{"qqq": [{"aaa": "333"}, {"bbb": "111"}], "zzz": {"bbb": [1, 2, 5, 9]}}'

现在我使用以下语句:

class ListEncoder(json.JSONEncoder):
    def default(self, o):
        if isinstance(o, list):
            return sorted(o)
        return json.JSONEncoder.default(self, o)
print json.dumps(c, sort_keys=True, cls=ListEncoder)

但是我的对象中的两个列表没有排序，我得到:

'{"qqq": [{"bbb": "111"}, {"aaa": "333"}], "zzz": {"bbb": [5, 2, 1, 9]}}'

可能是因为自定义JSONEncoder跳过了已经知道如何管理(list)的类型。

更新

下面的martinjn解决方案对于上面的示例非常有效，但不幸的是，我必须管理更复杂的字典，具有更大的深度:例如以下两个

a = {
    'aaa': 'aaa',
    'op': 'ccc',
    'oppa': {
        'ggg': [{'fff': 'ev'}],
        'flt': {
            'nnn': [
                {
                'mmm': [{'a_b_d': [6]},{'a_b_c': [6,7]}]
                },
                {
                    'iii': [3, 2, 4, 5]
                }
            ]
        }
    },
    'rrr': {},
    'ttt': ['aaa-bbb-ccc']
}
b = {
    'aaa': 'aaa',
    'op': 'ccc',
    'oppa': {
        'ggg': [{'fff': 'ev'}],
        'flt': {
            'nnn': [
                {
                    'iii': [2, 3, 4, 5]
                },
                {
                'mmm': [{'a_b_c': [6,7]},{'a_b_d': [6]}]
                }
            ]
        }
    },
    'rrr': {},
    'ttt': ['aaa-bbb-ccc']
}

如果同一序列中的列表被排序，则它们是相同的。但它们不是上面的类，我得到两个不同的json字符串:

{"aaa": "aaa", "op": "ccc", "oppa": {"flt": {"nnn": [{"iii": [3, 2, 4, 1]}, {"mmm": [{"a_b_d": [6]}, {"a_b_c": [6, 7]}]}]}, "ggg": [{"fff": "ev"}]}, "rrr": {}, "ttt": ["aaa-bbb-ccc"]}
{"aaa": "aaa", "op": "ccc", "oppa": {"flt": {"nnn": [{"iii": [2, 3, 4, 5]}, {"mmm": [{"a_b_c": [6, 7]}, {"a_b_d": [6]}]}]}, "ggg": [{"fff": "ev"}]}, "rrr": {}, "ttt": ["aaa-bbb-ccc"]}

有什么办法解决这个问题吗?

Answer 1

default不调用列表;该方法仅适用于编码器不知道如何处理的类型。重写encode方法:

class SortedListEncoder(json.JSONEncoder):
    def encode(self, obj):
        def sort_lists(item):
            if isinstance(item, list):
                return sorted(sort_lists(i) for i in item)
            elif isinstance(item, dict):
                return {k: sort_lists(v) for k, v in item.items()}
            else:
                return item
        return super(SortedListEncoder, self).encode(sort_lists(obj))

这实际上只是在编码之前对所有列表进行排序(递归地);这可以在将其传递给json.dumps()之前完成，但这种方式它是编码器责任的一部分，就像对密钥进行排序一样。

演示:

>>> json.dumps(c, sort_keys=True, cls=SortedListEncoder)
'{"qqq": [{"aaa": "333"}, {"bbb": "111"}], "zzz": {"bbb": [1, 2, 5, 9]}}'
>>> json.dumps(a, sort_keys=True, cls=SortedListEncoder)
'{"aaa": "aaa", "op": "ccc", "oppa": {"flt": {"nnn": [{"iii": [2, 3, 4, 5]}, {"mmm": [{"a_b_c": [6, 7]}, {"a_b_d": [6]}]}]}, "ggg": [{"fff": "ev"}]}, "rrr": {}, "ttt": ["aaa-bbb-ccc"]}'
>>> json.dumps(b, sort_keys=True, cls=SortedListEncoder)
'{"aaa": "aaa", "op": "ccc", "oppa": {"flt": {"nnn": [{"iii": [2, 3, 4, 5]}, {"mmm": [{"a_b_c": [6, 7]}, {"a_b_d": [6]}]}]}, "ggg": [{"fff": "ev"}]}, "rrr": {}, "ttt": ["aaa-bbb-ccc"]}'

Answer 2

更新后的问题应该是一个新问题，但我的更新解决方案是扩展已接受的答案，向列表排序添加一个更复杂的键:

class SortedListEncoder(json.JSONEncoder):
    def encode(self, obj):
        def get_key(item):
            if isinstance(item, dict):
                return get_key(sorted(item.keys()))
            else:
                return str(item)
        def sort_lists(item):
            if isinstance(item, list):
                return sorted((sort_lists(i) for i in item), key=lambda nm: get_key(nm))
            elif isinstance(item, dict):
                return {k: sort_lists(v) for k, v in item.items()}
            else:
                return item
        return super(SortedListEncoder, self).encode(sort_lists(obj))

，它允许字典根据键的排序列表进行比较。

这不是对象的完整排序，但它为您的两个测试用例(和我的)返回相同的排序:

{"aaa": "aaa", "op": "ccc", "oppa": {"flt": {"nnn": [{"iii": [2, 3, 4, 5]}, {"mmm": [{"a_b_c": [6, 7]}, {"a_b_d": [6]}]}]}, "ggg": [{"fff": "ev"}]}, "rrr": {}, "ttt": ["aaa-bbb-ccc"]}

它不能涵盖包含具有相同"first"的字典的列表的排序。关键但不同的值，如:

a=[{"bb": ["aa", "dd"]}, {"bb": ["cc", "dd"]}]
b=[{"bb": ["dd", "cc"]}, {"bb": ["dd", "aa"]}]

产生已排序的子列表，但保持字典顺序不变:

[{"bb": ["aa", "dd"]}, {"bb": ["cc", "dd"]}]
[{"bb": ["cc", "dd"]}, {"bb": ["aa", "dd"]}]

Answer 3

我把这个留在这里，因为我遇到了同样的问题。

您可以使用这个函数对嵌套的数据结构进行排序:

def sort_data(data):
    if isinstance(data, dict):
        output = OrderedDict()
        for key, value in data.items():
            output[key] = sort_data(value)
        return output
    elif isinstance(data, list):
        calculated = [sort_data(x) for x in data]
        return sorted(calculated, key=str)
    elif isinstance(data, (int, bool, str, float, type(None))):
        return data
    else:
        raise Exception("Unkown type: {} for {}".format(type(data), data))

的例子:

data = {"b":[ "zzz", "yyy", "xxx"],
        "d": [42, 54, 675, "aaa"],
        "c": {"a": ["bbb", "ccc", "aaa"]},
        }
sorted_data = sort_data(data)
print(json.dumps(sorted_data, indent=2, sort_keys=True))
# prints:
#{
#    "b": [
#        "xxx",
#        "yyy",
#        "zzz"
#    ],
#    "c": {
#        "a": [
#            "aaa",
#            "bbb",
#            "ccc"
#        ]
#    },
#    "d": [
#        42,
#        54,
#        675,
#        "aaa"
#    ]
#}