@pcdesign

Как пробежаться по стуктуре данных и получить все вложенные id?

Вот есть такая структура:
[
  {
    "id":1,
    "parent":""
  },
  {
    "id":2,
    "parent":1
  },
  {
    "id":3,
    "parent":1
  },
  {
    "id":4,
    "parent":3
  },
  {
    "id":5,
    "parent":3
  }
]


Как добавить список всех ID, которые являются вложенными, чтобы получилось вот так:

[
  {
    "id":1,
    "parent":"",
    "sub_ids": [2, 3, 4, 5]
  },
  {
    "id":2,
    "parent":1,
    "sub_ids": []
	
  },
  {
    "id":3,
    "parent":1,
    "sub_ids": [4, 5]	

  },
  {
    "id":4,
    "parent":3,
    "sub_ids": []
  },
  {
    "id":5,
    "parent":3,
    "sub_ids": []
  }
]
  • Вопрос задан
  • 117 просмотров
Решения вопроса 1
@pcdesign Автор вопроса
Вот решение с помощью рекурсии. Мне оно не нравится, но что-то другое я не могу придумать:

def get_sub_ids(parent, arr):
    for sub in my_list:
        if sub['parent'] == parent:
            arr.append(sub['id'])
            get_sub_ids(sub['id'], arr)

    return arr


for item in my_list:
    item['sub_ids'] = get_sub_ids(item['id'], [])


Результат:

[{'id': 1, 'parent': '', 'sub_ids': [2, 3, 4, 5]}, {'id': 2, 'parent': 1, 'sub_ids': []}, {'id': 3, 'parent': 1, 'sub_ids': [4, 5]}, {'id': 4, 'parent': 3, 'sub_ids': []}, {'id': 5, 'parent': 3, 'sub_ids': []}]
Ответ написан
Пригласить эксперта
Ответы на вопрос 1
Tomio
@Tomio
backend developer (python, php)
lst = [
  {
    "id":1,
    "parent":""
  },
  {
    "id":2,
    "parent":1
  },
  {
    "id":3,
    "parent":1
  },
  {
    "id":4,
    "parent":3
  },
  {
    "id":5,
    "parent":3
  }
]

for item in lst:
    item.update({'sub_ids': [sub['id'] for sub in lst if sub['parent'] == item['id']]})

print(lst)
Ответ написан
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы
06 авг. 2020, в 22:41
900 руб./в час
06 авг. 2020, в 22:18
3000 руб./за проект
06 авг. 2020, в 22:15
60000 руб./за проект