这里是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)
答案 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