python list 转 字典，父节点包含子节点

list 转 字典，父节点包含子节点

class Data:
    def __init__(self, id, p_id, name):
        self.id = id
        self.p_id = p_id
        self.name = name
        
def convert_to_dict(data_list):
    result_dict = {}
    for data in data_list:
        if data.p_id not in result_dict:
            result_dict[data.p_id] = [data.id]
        else:
            result_dict[data.p_id].append(data.id)
        child_dict = convert_to_dict(filter(lambda x: x.id == data.id, data_list))
        if len(child_dict) > 0:
            if data.p_id not in result_dict[data.id]:
                result_dict[data.p_id] = [data.id]
            else:
                result_dict[data.p_id].append(data.id)
            for child_key, child_value in child_dict.items():
                if child_key not in result_dict[data.p_id]:
                    result_dict[data.p_id].append(child_key)
                elif result_dict[data.p_id][-1] != child_key:
                    result_dict[data.p_id].append(child_key)
    return result_dict
    
# 示例用法
data_list = [Data(1, 2, 'Alice'), Data(2, 3, 'Bob'), Data(3, 4, 'Charlie')]
result_dict = convert_to_dict(data_list)
print(result_dict)