Выберите столбцы DataFrame на основе другого DataFrame

Question

Я пытаюсь выбрать подмножество DataFrame на основе столбцов другого DataFrame.

Фреймы данных выглядят так:

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

   a  b
0  0  1
1  2  3
2  4  5
3  6  7
4  8  9

Я хочу получить все строки первого Dataframe для столбцов, которые включены в оба DataFrames. Мой результат должен выглядеть так:

    a   b   
0   0   1   
1   4   5   
2   8   9  
3  12  13    

Answer 1

Вы можете использовать pd.Index.intersection или его синтаксический сахар &:

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

Answer 2

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)]