Compare column names of Pandas Dataframe

Question

How to compare column names of 2 different Pandas data frame. I want to compare train and test data frames where there are some columns missing in test Data frames??

Answer 1

pandas.Index objects, including dataframe columns, have useful set-like methods, such as intersection and difference.

For example, given dataframes train and test:

train_cols = train.columns
test_cols = test.columns

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