Сравните названия столбцов Pandas Dataframe

Question

Как сравнить имена столбцов 2 разных фреймов данных Pandas. Я хочу сравнить обучающие и тестовые фреймы данных, если в тестовых фреймах данных отсутствуют некоторые столбцы ??

Answer 1

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)