根据另一个数据框选择一个数据框的列

Question

我试图根据另一个DataFrame的列选择一个DataFrame的子集。

DataFrames看起来像这样：

    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

我想获取两个数据帧中包含的列的第一个数据帧的所有行。我的结果应如下所示：

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

Answer 1

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

Answer 2

您可以使用pd.Index.intersection或其语法糖&：

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