生成扁平化數據在Python

Question

嵌套列表生成內容的表，我有一個Python列表中的可用這些數據：

data = [ 
    {title: 'Section 1', level: 1, page_number: 1}, 
    {title: 'Section 1.1', level: 2, page_number: 2}, 
    {title: 'Section 1.2', level: 2, page_number: 3}, 
    {title: 'Section 2', level: 1, page_number: 4}, 
    {title: 'Section 2.1', level: 2, page_number: 5}, 
    {title: 'Section 3', level: 1, page_number: 6}, 
] 

由此，我想獲得這種嵌套結構的，與使用模板引擎更兼容：

toc = [ 
    {title: 'Section 1', page_number: 1, sub: [ 
     {title: 'Section 1.1', page_number: 2, sub: []}, 
     {title: 'Section 1.2', page_number: 3, sub: []}, 
    ]}, 
    {title: 'Section 2', page_number: 4, sub: [ 
     {title: 'Section 2.1', page_number: 5, sub: []},  
    ]}, 
    {title: 'Section 3', page_number: 6, sub: []}, 
] 

有關如何實現此目的的提示？我嘗試了遞歸函數，但對於我有限的大腦來說，它變得非常棘手。

任何幫助非常感謝。

編輯：增加了一個事實，一個條目最終可以沒有孩子。對不起，錯過了。

Answer 1

假設章節進來順序，這意味着孩子章總是父母后，並沒有缺失的父（跳過級）：

import pprint 

data = [ 
    {'title': 'Section 1', 'level': 1, 'page_number': 1}, 
    {'title': 'Section 1.1', 'level': 2, 'page_number': 2}, 
    {'title': 'Section 1.2', 'level': 2, 'page_number': 3}, 
    {'title': 'Section 2', 'level': 1, 'page_number': 4}, 
    {'title': 'Section 2.1', 'level': 2, 'page_number': 42}, 
    {'title': 'Section 2.1.1', 'level': 3, 'page_number': 42}, 
    {'title': 'Section 3', 'level': 1, 'page_number': 42}, 
] 

toc = [] 
stack = [toc] 
for d in data: 
    d['sub'] = [] 
    while d['level'] < len(stack): 
     stack.pop() 
    while d['level'] > len(stack): 
     stack.append(stack[-1][-1]['sub']) 
    stack[-1].append(d) 


pprint.pprint(toc) 

結果：

[{'level': 1, 
    'page_number': 1, 
    'sub': [{'level': 2, 'page_number': 2, 'sub': [], 'title': 'Section 1.1'}, 
      {'level': 2, 'page_number': 3, 'sub': [], 'title': 'Section 1.2'}], 
    'title': 'Section 1'}, 
{'level': 1, 
    'page_number': 4, 
    'sub': [{'level': 2, 
      'page_number': 42, 
      'sub': [{'level': 3, 
        'page_number': 42, 
        'sub': [], 
        'title': 'Section 2.1.1'}], 
      'title': 'Section 2.1'}], 
    'title': 'Section 2'}, 
{'level': 1, 'page_number': 42, 'sub': [], 'title': 'Section 3'}] 

編輯：改變它有沒有孩子的空'子'項目。在編輯歷史記錄中查看其他變體。

Answer 2

您可以遍歷列表中的每個條目並從那裏創建一個新列表。無論何時出現「Section x.y」，您都會在sub下添加它。

newData = [] 
curParent = None 
for d in data: 
    # child 
    if d['title'].find('.') > 0: 
    assert curParent # Make sure we have a valid parent dictionary 
    curParent['sub'].append({'title': d['title'], 'page_number': d['page_number']) 
    # parent 
    else: 
    curParent = {'title': d['title'], 'page_number': d['page_number'], 'sub': []} 
    newData.append(curParent) 

這應該工作2或3級，如果你需要更多比不同的方法可能會更好。此外，find（'。'）可能不適用於其他標題，但要麼使用級別字段（在您的示例中似乎是冗餘的）或正則表達式。

Answer 3

這是做你想做的嗎？

TITLE, LEVEL, PAGE_NUMBER, SUB = 'title', 'level', 'page_number', 'sub' 
data = [ 
    {TITLE: 'Section 1', LEVEL: 1, PAGE_NUMBER: 1}, 
    {TITLE: 'Section 1.1', LEVEL: 2, PAGE_NUMBER: 2}, 
    {TITLE: 'Section 1.1.1', LEVEL: 3, PAGE_NUMBER: 2}, 
    {TITLE: 'Section 1.2', LEVEL: 2, PAGE_NUMBER: 3}, 
    {TITLE: 'Section 2', LEVEL: 1, PAGE_NUMBER: 4}, 
    {TITLE: 'Section 2.1', LEVEL: 2, PAGE_NUMBER: 5}, 
] 

levels = [ { SUB: [] } ] 
for section in data: 
    section = dict(section) 
    current = section[LEVEL] 
    section[SUB] = [] 
    levels[current-1][SUB].append(section) 
    del levels[current:] 
    levels.append(section) 

toc = levels[0][SUB] 
from pprint import pprint 
pprint(toc)