如果不等于另一行熊猫则删除行

时间:2019-03-18 09:57:48

标签: pandas

这里是df:

col1    col2    col3
A   A   23
B   A   32
C   D   43
E   E   54

并且想法是仅保留col1和col2相等的行并获取:

col1    col2    col3
A   A   23
E   E   54

我尝试过:

for col1,col2 in zip(df1.col1, df1.col2): 
    if col1 != col2:
        df2.drop(col1)

1 个答案:

答案 0 :(得分:3)

您可以通过boolean indexing在两列中选择具有相同值的所有行:

df = df1[df1.col1 == df1.col2]
#alternative
#df = df1[df1.col1.eq(df1.col2)]
print (df)
  col1 col2  col3
0    A    A    23
3    E    E    54
相关问题