比较Pandas Dataframe的列名

时间:2018-05-06 19:31:24

标签: python pandas numpy machine-learning data-science

如何比较2个不同Pandas数据帧的列名。我想比较测试数据帧中缺少某些列的列车和测试数据帧吗?

2 个答案:

答案 0 :(得分:3)

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)

答案 1 :(得分:0)

train_column = train.columns
test_column = test.columns

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