python - 字典的嵌套列表

Question

我将 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

Answer 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 循环版本更复杂一些。

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