如何比较2个不同Pandas数据帧的列名。我想比较测试数据帧中缺少某些列的列车和测试数据帧吗?
答案 0 :(得分:3)
pandas.Index
个对象(包括数据框列)具有有用的set
类似方法,例如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)
答案 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)