如何比较两个不同Pandas数据框的列名?我想比较训练和测试数据框,其中测试数据框中缺少了一些列?
回答:
pandas.Index
对象,包括数据框的列,具有有用的 set
类方法,例如 intersection
和 difference
。
例如,给定数据框 train
和 test
:
train_cols = train.columnstest_cols = test.columnscommon_cols = train_cols.intersection(test_cols)train_not_test = train_cols.difference(test_cols)