我正在尝试遍历层次结构数据帧并将所有可能的路线记录到另一个数据帧中。这些路线可以具有可变的深度。
原始数据帧(df)。最高的列意味着父列中的值不是任何列的子项:
| 家长 | 孩子 | 最高 |
|---|---|---|
| A | 乙 | 1 |
| 乙 | C | 0 |
| 乙 | d | 0 |
| d | e | 0 |
最终目标数据框:
| 3级 | 2级 | 1级 | 0级 |
|---|---|---|---|
| A | 乙 | C | |
| A | 乙 | d | e |
这就是我目前拥有的
def search(parent):
for i in range(df.shape[0]):
if(df.iloc[i,0] == parent):
search(df.iloc[i,1])
for i in range(df.shape[0]):
if(df.iloc[i,2] == 1):
search(df.iloc[i,0])
Run Code Online (Sandbox Code Playgroud)
我能够浏览层次结构,但我不知道如何将其保存为我想要的格式。