比较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)