如何将DataFrame转换为嵌套字典，子元素和父元素中都有相同的列

Question

I have the dataframe as follows.

dictdata={'id':[991,992,989,5,4],'parentid':[4,4,4,0,0],'pcid': ['4_2','4_3','4_1','5','4']}
df = pd.DataFrame(dictdata)


   id   parentid  pcid
0  991         4  4_2
1  992         4  4_3
2  989         4  4_1
3    5         0    5
4    4         0    4

列id是父列，parentid表示id（行）中的

I want to generate the dictionary from above dataframe as follows:

{'data': [{'id': '5', 'level': 0, 'parentid': 0, 'pcid': '5', 'children': []},
          {'id': '4',
           'level': 0,
           'parentid': 0,
           'pcid': '4',
           'children': [{'id': '991',
                         'level': 2, 
                         'parentid': 4, 
                         'pcid': '4_2'
                        },
                        {'id': '992',
                         'level': 3,
                         'parentid': 4,
                         'pcid': '4_3'
                        },
                       {'id': '989', 
                        'level': 1, 
                        'parentid': 4,
                        'pcid': '4_1'
                       }
          ]}
        ]}

Answer 1

Create parent and child dataframe from actual dataframe

dfchild = df.loc[~(df['id']==df['pcid'])]
dfparent = df.loc[(df['id']==df['pcid'])] 

convert these df into dict:

parentdict = dfparent.to_dict('records')
childdict = dfchild.to_dict('records')

iterate the parentdict and inside get the respective child element from childdict based on parentid:

for parentdic in parentdict:
    eleid= int(parentdic['id'])
    childdictva=[]
    for child in childdict:
        childele=int(float(child['parentid']))
        if childele==eleid:
            childdictva.append(child)
    parentdic['children']=childdictva

Final code:

df['id']=df['id'].apply(str)
dfchild = df.loc[~(df['id']==df['pcid'])]
dfparent = df.loc[(df['id']==df['pcid'])]   
data = {}
parentdict = dfparent.to_dict('records')
childdict = dfchild.to_dict('records')
for parentdic in parentdict:
    eleid= int(parentdic['id'])
    childdictva=[]
    for child in childdict:
        childele=int(float(child['parentid']))
        if childele==eleid:
            childdictva.append(child)
    parentdic['children']=childdictva
data["data"] = parentdict

parentdict
{'data': [{'id': '5', 'level': 0, 'parentid': 0, 'pcid': '5', 'children': []},
 {'id': '4',
 'level': 0,
 'parentid': 0,
 'pcid': '4',
 'children': [{'id': '991', 'level': 2, 'parentid': 4, 'pcid': '4_2'},
  {'id': '992', 'level': 3, 'parentid': 4, 'pcid': '4_3'},
  {'id': '989', 'level': 1, 'parentid': 4, 'pcid': '4_1'}]}]}