Вложенные списки словарей - PullRequest
0 голосов
/ 06 мая 2018

У меня есть 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 голосов
/ 06 мая 2018

Я чувствую, что ваш подход терпит неудачу, потому что вы пытаетесь сделать все сразу: это однострочное назначение довольно нетривиально. Рассмотрим вместо этого что-то вроде этого:

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 работать только на определенном уровне вашей структуры и не заботиться о общей картине.

Примечание: я не проверял это, потому что у меня нет доступа к данным, отформатированным как у вас (с атрибутами и итерациями и всем прочим).

Подумав некоторое время, я вижу, что это можно легко переписать как огромный список. Является ли это более читабельным, однако сомнительно. Он выглядит намного чище и напоминает целевую структуру по форме, но он несколько сложнее, чем простая версия для цикла.

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]
}
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...