python 实现一个求解当前数据所有子项的方法

Question

假设数据结构是这样的：
有多条数据，每条数据都有属性parent(指向它的所属上级id)，自身的唯一标识id。

class data
    parent
    id

当我拿出其中一条数据的时候，用一个方法计算出所有下级数据。就好比中国行政区数据一样，当我拿到广东省的时候，下面的市级，县级都要得到，如果县级下面还有分级也要一并拿到。

我写了一些代码，都太丑陋了，似乎是要用递归，求教一下有没有什么好的思路？

Answer 1

这个问题可以转换成N叉树的遍历，将某个节点的所有子节点计算出来。

    def return_all_children(self):
        """
        返回所有的子项
        :return:
        """
        root = self
        if not root:
            return []
        que = []  # 保存节点的队列
        res = []  # 保存结果的列表
        que.append(root)  #
        while len(que):  # 判断队列不为空
            length = len(que)
            sub = []  # 保存每层的节点的值
            for i in range(length):
                current = que.pop(0)  # 出队列的当前节点
                print(current)
                sub.append(current)
                for child in current.return_children:  # 所有子结点入队列
                    que.append(child)
            res.append(sub)  # 把每层的节点的值加入结果列表
        return res

Answer 2

储存空间够大的话可以建立一个字典
假如你的数据是个list名字叫CN（中国所有省市县...）

parent_d = {}
for item in CN:
    parent = item['parent']
    if parent in parent_d:
        parent_d[parent].append(item['id'])
    else:
        parent_d[parent] = [item['id']]

之后遍历一下

def get_all_children(city_id, parent_d):
    if city_id not in parent_d or not city_id:
        return []
    result = []
    temp_parent = [city_id]
    while temp:
        cur_id = temp_parent.pop(-1)
        result += parent_d[cur_id]
        temp_parent = parent_d[cur_id] + temp_parent
    return result

Answer 3

贴一下全一点的例子