如何自定义要在json.dumps中使用的dict列表

时间:2013-09-18 11:37:54

标签: python json list sorting dictionary

我有一个类似于

的列表
allsites = [
    {
        'A5': 'G', 
        'A10': 'G', 
        'site': 'example1.com', 
        'A1': 'G'
    }, 
    {
        'A5': 'R', 
        'A10': 'Y',
        'site': 'example2.com', 
        'A1': 'G'
    }
]

我在json.dumps中使用

data = { 'Author':"joe", 'data':allsites }
print json.dumps(data,sort_keys=True,indent=4, separators=(',', ': '))

这会输出以下JSON:

{
    "Author": "joe",
    "data": [
        {
            "A1": "G",
            "A10": "G",
            "A5": "G",
            "site": "example1.com"
        },
        {
            "A1": "G",
    (...)

我希望通过自定义键(“alphabet”)对此JSON字符串的“data”部分进行排序,在上面的情况下,这将是site, A1, A5, A10,实际上看起来像:

{
    "Author": "joe",
    "data": [
        {
            "site": "example1.com",
            "A1": "G",
            "A5": "G",
            "A10": "G"
        },
        {
            "site": "example2.com",
            "A1": "G",
    (...)

我在Sorting FAQ中读到了自定义排序,但它只是提供了一种覆盖比较函数的方法,更不用说我不知道​​如何将它插入到我的代码中。

怎么做?

3 个答案:

答案 0 :(得分:23)

由于python dicts是无序集合,因此请使用collections.OrderedDict自定义排序:

from collections import OrderedDict
import json

allsites = [
    {
        'A5': 'G',
        'A10': 'G',
        'site': 'example1.com',
        'A1': 'G'
    },
    {
        'A5': 'R',
        'A10': 'Y',
        'site': 'example2.com',
        'A1': 'G'
    }
]

sort_order = ['site', 'A1', 'A5', 'A10']
allsites_ordered = [OrderedDict(sorted(item.iteritems(), key=lambda (k, v): sort_order.index(k)))
                    for item in allsites]

data = {'Author': "joe", 'data': allsites_ordered}
print json.dumps(data, indent=4, separators=(',', ': '))

打印:

{
    "data": [
        {
            "site": "example1.com",
            "A1": "G",
            "A5": "G",
            "A10": "G"
        },
        {
            "site": "example2.com",
            "A1": "G",
            "A5": "R",
            "A10": "Y"
        }
    ],
    "Author": "joe"
}

答案 1 :(得分:8)

在Python3中,alecxe的答案不再适用。这应该是一个评论,但我缺乏声誉。

PEP 3113删除了函数签名中的元组解包,所以行

allsites_ordered = [OrderedDict(sorted(item.iteritems(), key=lambda (k, v): sort_order.index(k)))
                    for item in allsites]

现在必须是

allsites_ordered = [OrderedDict(sorted(item.items(), key=lambda item: sort_order.index(item[0])))
                    for item in allsites]

或类似的。 iteritems也变为items

答案 2 :(得分:5)

我遇到了完全相同的问题并设计了一个轻量级的通用解决方案:

from collections import OrderedDict

def make_custom_sort(orders):
    orders = [{k: -i for (i, k) in enumerate(reversed(order), 1)} for order in orders]
    def process(stuff):
        if isinstance(stuff, dict):
            l = [(k, process(v)) for (k, v) in stuff.items()]
            keys = set(stuff)
            for order in orders:
                if keys.issuperset(order):
                    return OrderedDict(sorted(l, key=lambda x: order.get(x[0], 0)))
            return OrderedDict(sorted(l))
        if isinstance(stuff, list):
            return [process(x) for x in stuff]
        return stuff
    return process

首先,您创建一个自定义顺序排序函数的实例:

custom_sort = make_custom_sort([ ["site", "A1", "A5", "A10"] ])

现在,实际排序:

result = custom_sort(allsites)

...您可以将其转储为JSON对象:

print json.dumps(result, indent=4)

结果

[
    {
        "site": "example1.com", 
        "A1": "G", 
        "A5": "G", 
        "A10": "G"
    }, 
    {
        "site": "example2.com", 
        "A1": "G", 
        "A5": "R", 
        "A10": "Y"
    }
]

更多

闭包是递归的。如双括号所示,您可以指定与结构中嵌套的各种词典一样多的排序顺序。

GitHub项目:https://github.com/laowantong/customsort