Compare Two Columns in Two Pandas Dataframes

Question

I have two pandas dataframes:

df1:
a   b   c
1   1   2
2   1   2
3   1   3

df2:
a   b   c 
4   0   2 
5   5   2 
1   1   2 

df1 = {'a': [1, 2, 3], 'b': [1, 1, 1], 'c': [2, 2, 3]}
df2 = {'a': [4, 5, 6], 'b': [0, 5, 1], 'c': [2, 2, 2]}
df1= pd.DataFrame(df1)
df2 = pd.DataFrame(df2)

I'm looking for a function that will display whether df1 and df2 contain the same value in column a.

In the example I provided df1.a and df2.a both have a=1.

If df1 and df2 do not have an entry where the the value in column a are equal then the function should return None or False.

How do I do this? I've tried a couple combinations of panda.merge

BENY · Accepted Answer · 2019-03-30 01:52:11Z

1

Define your own function by using isin and any

def yourf(x,y):
    if any(x.isin(y)):
        #print(x[x.isin(y)])
        return x[x.isin(y)]
    else: 
        return 'No match' # you can change here to None

Out[316]: 
0    1
Name: a, dtype: int64

yourf(df1.b,df2.c)
Out[318]: 'No match'

answered Mar 30, 2019 at 1:52

BENY

324k22 gold badges176 silver badges250 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

busybear · Accepted Answer · 2019-03-30 02:01:59Z

1

You could use set intersection:

def col_intersect(df1, df2, col='a'):
    s1 = set(df1[col])
    s2 = set(df2[col])
    return s1 & s2 else None

Using merge as you tried, you could try this:

def col_match(df1, df2, col='a'):
    merged = df1.merge(df2, how='inner', on=col)
    if len(merged):
        return merged[col]
    else:
        return None

edited Mar 30, 2019 at 2:01

answered Mar 30, 2019 at 1:50

busybear

10.7k1 gold badge29 silver badges44 bronze badges

Collectives™ on Stack Overflow

Compare Two Columns in Two Pandas Dataframes

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related