比较 Pandas 数据框的列名

Question

如何比较 2 个不同 Pandas 数据框的列名。我想比较训练和测试数据帧，其中测试数据帧中缺少一些列？？

Answer 1

pandas.Index objects, including dataframe columns, have useful set-like methods, such as intersection and 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)

Answer 2

train_column = train.columns
test_column = test.columns

common_column = train_column.intersection(test_column)
train_not_in_test = train_column.difference(test_column)