递归迭代列表

Question

我正在解析JSON并尝试基于“parents”键递归地创建一个树，但是我很难获得我期望的输出。以下是JSON：

{
    "cases": [
        {
            "name": "A",
            "parents": [
                "E",
                "B"
            ]
        },
        {
            "name": "B",
            "parents": [
                "C",
                "D"
            ]
        },
                {
            "name": "C",
            "parents": [
                "E"
            ]
        },
        {
            "name": "D",    
            "parents": [
                "E"
            ]
        },
        {
            "name": "E",
            "parents": [
            ]
        }
    ]
}

这是我的密码：

import json
import copy
FILE="somefile.json"

def find_path(test_name, data, current_steps):
    tc = return_test_data(test_name, data)
    parents = tc.get('parents', [])
    if not current_steps:
        current_steps.append(test_name)
    if not parents:
        return current_steps
    else:
        temp_steps = []
        for step in parents:
            new_c_s = copy.deepcopy(current_steps)
            new_c_s.append(step)
            new_steps = find_path(step, data, new_c_s)    
            temp_steps.append(new_steps)
        return temp_steps


def return_test_data(test_name, data):
    for test in data['cases']:
        if test['name'] == test_name:
            return test

    return False


if __name__ == "__main__":
    data = json.load(open(FILE))

    steps = find_path("A", data, [])
    print ("Steps: {}".format(steps))

我希望能看到一个简单的父母名单，按他们被找到的顺序排列：

Steps: ['A', 'E', 'B', 'C', 'E', 'D', 'E']

在本例中，A有两个父对象：E&B。迭代这两个对象以获取它们的父对象，因此结果将是：

['A', 'E', {parents of E}, 'B', {parents of B}].

因为E没有父母，B有C和D的父母，这（在我看来）变成：

['A', 'E', 'B', 'C', {parents of C}', 'D', {parents of D}]

最终变成：

['A', 'E', 'B', 'C', 'E', 'D', 'E']

相反，我得到：

Steps: [['A', 'E'], [[['A', 'B', 'C', 'E']], [['A', 'B', 'D', 'E']]]]

我肯定我传递了太多的递归，但不知道到底是什么。你知道吗

Answer 1

您可以像这样迭代结构：

代码：

parents_list = []
for record in data:
    if record['name'] not in parents_list:
        parents_list.append(record['name'])
        for p in record['parents']:
            parents_list.append(p)

print(parents_list)

结果：

['A', 'E', 'B', 'C', 'E', 'D', 'E']

Answer 2

看起来你把每一步都做得比必要的稍微复杂一点。我相信这会产生你想要的信息：

import json

FILE = "somefile.json"

def find_path(test_name, data):

    dictionary = return_test_data(test_name, data)

    if dictionary:
        current_steps = [test_name]

        if 'parents' in dictionary:
            for parent in dictionary['parents']:
                current_steps.extend(find_path(parent, data))

        return current_steps

    return None

def return_test_data(name, data):
    for dictionary in data['cases']:
        if dictionary['name'] == name:
            return dictionary

    return None

if __name__ == "__main__":
    data = json.load(open(FILE))

    steps = find_path("A", data)

    print("Steps:", steps)

输出

> python3 test.py
Steps: ['A', 'E', 'B', 'C', 'E', 'D', 'E']
>