python - RDD映射创建层次树

Question

如何构造一个 table/rdd 以便它描述来自 parent 的级别。下面给出的是“：”之后的节点及其对应的子节点

1: 3,4
2: 1,3
3: 6
4: 5
5:
6: 5

对于上表，如果我的源输入为 2，目标输入为 5，iIconstruct 如何构造一个 rdd 使其看起来像

2 : null 0
1: 2 1
3: 2 1
4: 1 2
3: 1 2
6: 3 2
5: 4 3

其中：旁边的数字表示父节点，下一个数字表示根节点的级别。