Select Columns of a DataFrame based on another DataFrame

Question

I am trying to select a subset of a DataFrame based on the columns of another DataFrame.

The DataFrames look like this:

    a   b   c   d
0   0   1   2   3
1   4   5   6   7
2   8   9  10  11
3  12  13  14  15

I want to get all rows of the first Dataframe for the columns which are included in both DataFrames. My result should look like this:

jpp · Accepted Answer · 2018-09-04 09:49:00Z

6

You can use pd.Index.intersection or its syntactic sugar &:

intersection_cols = df1.columns & df2.columns
res = df1[intersection_cols]

answered Sep 4, 2018 at 9:49

jpp

166k37 gold badges301 silver badges362 bronze badges

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

Comments

ram nithin · Accepted Answer · 2018-09-04 10:04:20Z

3

import pandas as pd

data1=[[0,1,2,3,],[4,5,6,7],[8,9,10,11],[12,13,14,15]]
data2=[[0,1],[2,3],[4,5],[6,7],[8,9]]

df1 = pd.DataFrame(data=data1,columns=['a','b','c','d'])
df2 = pd.DataFrame(data=data2,columns=['a','b'])

df1[(df1.columns) & (df2.columns)]

edited Sep 4, 2018 at 10:04

answered Sep 4, 2018 at 9:47

ram nithin

1195 bronze badges

Collectives™ on Stack Overflow

Select Columns of a DataFrame based on another DataFrame

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related