比较熊猫数据框的列名

pandas.Index 对象(包括dataframe列)具有有用的 set -like方法,例如 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)

train_column = train.columns
test_column = test.columns

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