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

2024-05-29 03:30:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个类似于

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",
    (...)

我希望这个JSON字符串的“data”部分通过一个自定义键(“alphabet”)进行排序,在上面的例子中是site, A1, A5, A10,实际上看起来像:

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

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

怎么做?


Tags: comjsondata排序a1sitea10author
3条回答

由于python dict是无序的集合,因此请将^{}与自定义排序一起使用:

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"
}

我也遇到了同样的问题,并设计了一个轻量级通用解决方案:

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

在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

相关问题 更多 >

    热门问题