Comparing 2 columns of two Python Pandas dataframes and getting the common rows

前端 未结 4 1013
旧巷少年郎
旧巷少年郎 2020-12-03 18:43

I have 2 Dataframe as follows:

DF1=
    A    B   C    D
0   AA   BA  KK   0
1   AD   BD  LL   0
2   AF   BF  MM   0

DF2=
    K    L
0   AA   BA
1   AD   BF
         


        
4条回答
  •  星月不相逢
    2020-12-03 19:47

    This is how I solved it:

    df1 = pd.DataFrame({"A":['AA','AD','AD'], "B":['BA','BD','BF']})
    df2 = pd.DataFrame({"A":['AA','AD'], 'B':['BA','BF']})
    df1['compressed']=df1.apply(lambda x:'%s%s' % (x['A'],x['B']),axis=1)
    df2['compressed']=df2.apply(lambda x:'%s%s' % (x['A'],x['B']),axis=1)
    df1['Success'] = df1['compressed'].isin(df2['compressed']).astype(int)
    print df1
    
        A   B     compressed   Success
    0  AA  BA      AABA          1
    1  AD  BD      ADBD          0
    2  AD  BF      ADBF          1
    

提交回复
热议问题