處理遞回Python函式中的作用域

我有一個如下所示的 JSON 輸入檔案：

{"nodes": [
    {"properties": {
        "id": "rootNode",
        "name": "Bertina Dunmore"},
      "nodes": [
        {"properties": {
            "id": 1,
            "name": "Gwenneth Rylett",
            "parent_id": "rootNode"},
          "nodes": [
            {"properties": {
                "id": 11,
                "name": "Joell Waye",
                "parent_id": 1}},
            {"properties": {
                "id": 12,
                "name": "Stan Willcox",
                "parent_id": 1}}]},
        {"properties": {
            "id": 2,
            "name": "Delbert Dukesbury",
            "parent_id": "rootNode"},
          "nodes": [
            {"properties": {
                "id": 21,
                "name": "Cecil McKeever",
                "parent_id": 2}},
            {"properties": {
                "id": 22,
                "name": "Joy Obee",
                "parent_id": 2}}]}]}]}

我想將嵌套properties字典放入（平面）字典串列中。創建將讀取此字典的遞回函式很容易：

def get_node(nodes):
    for node in nodes:
        print(node['properties'])
        if 'nodes' in node.keys():
            get_node(node['nodes'])

現在，我正在努力將這些附加到一個串列中：

def get_node(nodes):
    prop_list = []
    for node in nodes:
        print(node['properties'])
        prop_list.append(node['properties'])
        if 'nodes' in node.keys():
            get_node(node['nodes'])
    return prop_list

[{'id': 'rootNode', 'name': 'Bertina Dunmore'}]即使properties列印了所有字典，這也會回傳。我懷疑這是因為我沒有正確處理函式范圍。

有人可以幫我解決這個問題嗎？

uj5u.com熱心網友回復：

您需要將prop_list遞回呼叫回傳的值與prop_list當前范圍內的 結合起來。例如，

def get_node(nodes):
    prop_list = []
    for node in nodes:
        print(node['properties'])
        prop_list.append(node['properties'])
        if 'nodes' in node.keys():
            prop_list.extend(get_node(node['nodes']))
    return prop_list

uj5u.com熱心網友回復：

您的問題是每次呼叫 get_node 時，您附加的串列都會再次初始化。您可以通過將串列傳遞到遞回函式中來避免這種情況

此外，我認為使用資料類來處理這個問題會很好，

from dataclasses import dataclass
from typing import Union


@dataclass
class Property:
    id: int
    name: str
    parent_id: Union[str, None] = None
    
        
def explore_json(data, properties: list=None):
    if properties is None:
        properties = []
    for key, val in data.items():
        if key == "nodes":
            for node in val:
                explore_json(node, properties)
        elif key == "properties":
            properties.append(Property(**val))
    return properties

explore_json(data)

輸出

[Property(id='rootNode', name='Bertina Dunmore', parent_id=None),
 Property(id=1, name='Gwenneth Rylett', parent_id='rootNode'),
 Property(id=11, name='Joell Waye', parent_id=1),
 Property(id=12, name='Stan Willcox', parent_id=1),
 Property(id=2, name='Delbert Dukesbury', parent_id='rootNode'),
 Property(id=21, name='Cecil McKeever', parent_id=2),
 Property(id=22, name='Joy Obee', parent_id=2)]

uj5u.com熱心網友回復：

接著就，隨即：

def get_node(prop_list, nodes):
    for node in nodes:
        print(node['properties'])
        prop_list.append(node['properties'])
        if 'nodes' in node.keys():
            get_node(prop_list, node['nodes'])

你可以這樣做：

prop_list = []
get_node(prop_list, <yourdictnodes>)

應prop_list改為：

{'id': 'rootNode', 'name': 'Bertina Dunmore'}
{'id': 1, 'name': 'Gwenneth Rylett', 'parent_id': 'rootNode'}
{'id': 11, 'name': 'Joell Waye', 'parent_id': 1}
{'id': 12, 'name': 'Stan Willcox', 'parent_id': 1}
{'id': 2, 'name': 'Delbert Dukesbury', 'parent_id': 'rootNode'}
{'id': 21, 'name': 'Cecil McKeever', 'parent_id': 2}
{'id': 22, 'name': 'Joy Obee', 'parent_id': 2}

標籤：Python json 递归 范围

上一篇：嵌套串列深度python的總和

下一篇：Concat有效，但在展平陣列時push無效