我是如何循环检查df中每行的值以检查该值是否满足条件的

Question

我的表格如下：

ID   IsCreated  ParentID
101  0          NA
102  1          101
103  0          NA
104  1          102
105  1          104

我想添加一个名为“OGParentID”的新列，其结果如下所示：

ID   IsCreated  ParentID   OGParentID
101  0          NA         101        
102  1          101        101        
103  0          NA         103        
104  1          102        101        
105  1          104        101  

  

这里使用的逻辑如下： 对于每一行，检查IsCreated=1，如果是，则在ID列中查找ParentID以检查IsCreated是否为0，如果是，则将OGParentID设置为已查找的父ID，否则，检查已查找的ParentID的父ID并继续循环

Answer 1

你可以做：

#creating a dictionary "x" to have all the unique parent-child relationships using comprehension
x={i:int(j) for i,j in zip(df['ID'],df['ParentID'].fillna(df['ID']))}

#assigning ID to the new column 
df['OGParentID']=df['ID']

#loop to replace the dict items ,i.e, replacing all the childs with parents
#counter is limited to the length of unique elements in dict
i=0
while i<=len(x):
    df['OGParentID']=df['OGParentID'].replace(x)
    i+=1


Output: 

        ID  IsCreated  ParentID  OGParentID
    0  101          0       NaN         101
    1  102          1     101.0         101
    2  103          0       NaN         103
    3  104          1     102.0         101
    4  105          1     104.0         101