嵌套的词典列表

时间:2018-05-06 14:23:42

标签: python json python-2.7

我有3个对象,我填充为JSON文件格式。这些对象来自API,需要推出才能访问:

my_dict = {}
for elem_a in list_a:
    for elem_b in elem_a:
        for elem_c in elem_b:
            elem_c_info = {
                "name" : elem_c.prop1,
                "ID" : elem_c.prop2,
                "GPA" : elem_c.prop3
            }

            my_dict.setdefault("university", {}) \
                 .setdefault(str(elem_a), {}) \
                 .setdefault(str(elem_b), {}) \
                 .setdefault("student", []).append(elem_c_info)

生成如下输出:

{
    "university": {
        "universityA": {
            "class_1": {
                "student": [
                    {
                        "name": "student_1", 
                        "ID": "1234", 
                        "GPA": "3.8"
                        },

我想要的输出:

{
    "university": [{
        "name": "universityA",
        "class": [{
                "name": "class_1",
                "student": [{
                        "name": "student_1",
                        "ID": "1234",
                        "GPA": "3.8"
                    },
                    {
                        "name": "student_2",
                        "ID": "12345",
                        "GPA": "3.4"
                    }
                ]
            },
            {
                "name": "class_2",
                "student": [{
                    "name": "student_3",
                    "ID": "14",
                    "GPA": "3.0"
                }]
            }
        ]
    }]
}

正如您所看到的,我需要在每个级别的列表中包含每个嵌套字典,我需要在每个级别添加一个键/值对(除了最内层)。我成功地使用.append处理最内层的巢,但它没有在更高的位置上工作。有关将这些元素添加到数据结构的任何建议吗?接受任何一个或两个问题的解决方案。

这是输入数据结构看起来像运行一个简单的嵌套循环(感谢@ PM_2Ring建议):

在上面的例子中:

list_a = [obj_1, obj_2, ..., obj_n]  

for elem_a in list_a:
    print('A', elem_a.name)
    for elem_b in elem_a:
        print('  B', elem_b.name)
        for elem_c in elem_b:
            print('    C', elem_c.prop1, elem_c.prop2, elem_c.prop3)

('A', universityA)
('  B', class_1)
('    C', student_1, 12345, 3.8)
('  B', class_2)
('    C', student_2, 145, 3.6)
('A', universityB)
('  B', class_1)
('    C', student_1, 12345, 3.8)
('    C', student_2, 1235, 3.6)
('    C', student_3, 12345, 3.4)
('  B', class_2)
('    C', student_1, 145, 3.6)
  ....

elem_a = [universityA, universityB, universityC]  # Top tier
elem_b = [universityA.class_1, universityA.class_2, universityA.class_3]              # Middle tier with OO property 'name'
elem_c = [universityA.class_1.student_name, universityA.class_1.student_id, universityA.class_1.student_gpa]  # bottom tier relationship

1 个答案:

答案 0 :(得分:1)

我觉得你的方法失败了,因为你试图一次完成所有事情:单行任务是非常重要的。请考虑这样的事情:

my_dict = {}
list_result = []
for elem_a in list_a:
    elem_a_result = []
    for elem_b in elem_a:
        elem_b_result = []
        for elem_c in elem_b:
            elem_b_result.append({
                'prop1': elem_c.prop1,
                'prop2': elem_c.prop2,
                'prop3': elem_c.prop3,
            })
        elem_a_result.append({
            'name': elem_b.name,
            'elem_b': elem_b_result,
        })
    list_result.append({
        'name': elem_a.name,
        'elem_a': elem_a_result
    })
my_dict = {
    'list_a': list_result
}

您可以使每个for循环中的代码仅在您的结构的特定级别上工作,而不关心大局。

注意:我没有对此进行测试,因为我无法访问与您的格式相似的数据(包含属性和迭代以及所有内容)。

在思考了一段时间后,我发现这可以很容易地重写为一个巨大的列表理解。然而,这是否更具可读性是值得怀疑的。它看起来更清晰,类似于目标结构的形状,但它比简单的for-loop版本更复杂。

my_dict = {
    'list_a': [
        {
            'name': elem_a.name,
            'elem_a': [
                {
                    'name': elem_b.name,
                    'elem_b': [
                        {
                            'prop1': elem_c.prop1,
                            'prop2': elem_c.prop2,
                            'prop3': elem_c.prop3,
                        }
                    for elem_c in elem_b]
                }
            for elem_b in elem_a]
        }
    for elem_a in list_a]
}