比較Pandas Dataframe的列名

Question

如何比較2個不同的Pandas數據框的列名。 我想比較訓練數據幀和測試數據幀，其中測試數據幀中缺少某些列？

Answer 1

pandas.Index對象（包括數據pandas.Index列）具有有用的類似於set的方法，例如intersection和difference 。

例如，給定的數據幀train和test ：

train_cols = train.columns
test_cols = test.columns

common_cols = train_cols.intersection(test_cols)
train_not_test = train_cols.difference(test_cols)