两个pandas数据帧中共有的列列表

Question

我考虑对每个都有大量列的数据帧进行合并操作。不希望结果有两个具有相同名称的列。我试图查看两个框架之间共同的列名列表：

{{1}}

如何在Index对象上操作NumPy布尔数组，以便它只返回一个共同列的列表？

Answer 1

使用numpy.intersect1d或intersection：

a = np.intersect1d(df2.columns, df1.columns)
print (a)
['B' 'C']

a = df2.columns.intersection(df1.columns)
print (a)
Index(['B', 'C'], dtype='object')

后一种选择的替代语法：

df1.columns & df2.columns