python - 如何在父子关系上组合两个数据帧(介于连接和合并之间)

Question

我正在努力形成一个像数据框一样的树，其中子行就在其 parent 的下方。我想做的是合并 object_id xparent_id 和沿轴 0 连接之间的事情。 所以我正在寻找的是下面代码片段中交错函数的实现。

In[1]: parents = pd.DataFrame({'object_id':[1,2],
                               'parent_id':[0,0],
                               'position': [1,2]})

In[2]: parents

Out[2]    object_id     parent_id   position
       0  1             0           1
       1  2             0           2

In[3]: children = pd.DataFrame({'object_id':[3,4,5],
                                'parent_id':[1,1,2],
                                'position': [1,2,1]})

In[4]: children

Out[4]:   object_id     parent_id   position
       0  3             1           1
       1  4             1           2
       2  5             2           1

In[5]: interlace(parent, children, on=('object_id', 'parent_id'))

Out[5]:  object_id  parent_id   position
      0  1          0           1
      1  3          1           1
      2  4          1           2
      3  2          0           1
      4  5          2           1

有没有一种有效的方法可以在 pandas 中做到这一点？ 我认为人们可以做类似的事情

parents_with_children = []
for i, parentrow in parents.iteritems():
    childrenrows = children[children.parent_id == parentrow.object_id]
    parents_with_children.append(pd.concat([parentrow, childrenrows])
result = pd.concat(parents_with_children)

但我觉得应该有更简单、可能更有效的方法来做到这一点。

编辑:具有相同级别和相同父级的行需要按其位置排序。

Answer 1

可能的解决方案:

children['sort_id']=children.parent_id
parents['sort_id']=parents.object_id
pd.concat([parents,children]).sort_values(['sort_id', 'parent_id']).drop('sort_id', 1)