比较Pandas Dataframe的列名

Jaq*_*har 5 python numpy machine-learning pandas data-science

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

jpp*_*jpp 6

pandas.Index对象(包括数据框列)具有set类似类的有用方法,例如intersectiondifference

例如,给定的数据帧traintest

train_cols = train.columns
test_cols = test.columns

common_cols = train_cols.intersection(test_cols)
train_not_test = train_cols.difference(test_cols)
Run Code Online (Sandbox Code Playgroud)