Question

我有一个数据框，其中某些行具有父子关系。例如1002是1003的父级，1003是1004的父级

    import pandas as pd
    import numpy as np

    df = pd.DataFrame(columns=['Id', 'Parent Id', 'Child Id', 'A', 'B'],
                      data=[[1001, np.nan, 1005, 'A1001', 'B1001'],
                            [1002, np.nan, 1003, 'A1002', 'B1002'],
                            [1003, 1002, 1004, 'A1003', np.nan],
                            [1004, 1003, np.nan, 'A1004', np.nan],
                            [1005, 1001, np.nan, 'A1005', np.nan]
                            ])
    print(df)

         Id  Parent Id  Child Id      A      B
    0  1001        NaN    1005.0  A1001  B1001
    1  1002        NaN    1003.0  A1002  B1002
    2  1003     1002.0    1004.0  A1003    NaN
    3  1004     1003.0       NaN  A1004    NaN
    4  1005     1001.0       NaN  A1005    NaN

我想合并行以仅保留那些没有子项的行。我了解我可以使用

    df = df[df['Child Id'].isnull()]

但是，某些信息仅出现在父行中，而子行中却没有，例如列1002中的“ B”列。我想将其继承到最新的子行。

以下是预期的输出：

        Id  Parent Id  Child Id      A      B
    3  1004     1003.0       NaN  A1004  B1002
    4  1005     1001.0       NaN  A1005  B1001

代替迭代行，我可以使用内置的pandas方法吗？

Answer 1

您需要获取父子关系链中B列中的值。您可以执行以下操作：

for i in range(max_length_of_relationships):
    df = df.merge(df[['Id', 'B']].rename({'Id': 'Parent Id', 'B': 'Parent B'}, axis=1), how='left')
    df['B'] = df['B'].combine_first(df['Parent B'])
    df.drop('Parent B', axis=1, inplace=True)

其中max_length_of_relationships是一条链中的父子链接的最大数目（在df中为2：1。10002-10003，2.10003-10004），如果不确定该数目是多少，请使用无法超越的大量数字。然后像您一样，只保留没有孩子的行：

df = df[df['Child Id'].isnull()]

根据父ID合并同一数据帧中的多行

1 个答案: