在熊猫中创建层次结构列
我有一个这样的数据框:
part part_parent
0 part1 NaN
1 part2 part1
2 part3 part2
3 part4 part3
4 part5 part2
我需要像这样添加一个额外的列层次结构:
part part_parent hierarchy
0 part1 NaN part1
1 part2 part1 part1/part2/
2 part3 part2 part1/part2/part3/
3 part4 part3 part1/part2/part3/part4
4 part5 part2 part1/part2/part5
字典创建输入/输出数据帧:
from numpy import nan
df1 = pd.DataFrame({'part': {0: 'part1', 1: 'part2', 2: 'part3', 3: 'part4', 4: 'part5'},
'part_parent': {0: nan, 1: 'part1', 2: 'part2', 3: 'part3', 4: 'part2'}})
df2 = pd.DataFrame({'part': {0: 'part1', 1: 'part2', 2: 'part3', 3: 'part4', 4: 'part5'},
'part_parent': {0: nan, 1: 'part1', 2: 'part2', 3: 'part3', 4: 'part2'},
'hierarchy': {0: 'part1',
1: 'part1/part2/',
2: 'part1/part2/part3/',
3: 'part1/part2/part3/part4',
4: 'part1/part2/part5'}})
注意:我已经看到了几个与NetworkX解决此问题相关的线程,但我无法这样做。
任何帮助表示赞赏。