迭代检查列名是否等于另一列的值

时间:2018-01-23 18:05:54

标签: pandas loops

有没有办法迭代数据框,如果列名等于另一列的值,则在该特定单元格中输入“1”。

例如:

[A]  [B]  [C]  
D
E
B
C

会转换为:

[A] [B] [C]
D
E
B   1   
C       1

B和c中的空值只是NaN。

1 个答案:

答案 0 :(得分:1)

这样做

df.fillna('').apply(lambda x : x.index==x.name).astype(int).replace(0,"")
Out[563]: 
      A  B  C
index        
D            
E            
B        1   
C           1