Объединить два кадра данных по индексу и вернуть уникальные столбцы панд

Question

Я хочу объединить два фрейма данных в индексе и вернуть только отдельные столбцы, присутствующие после слияния.

В настоящее время я использую - pd.merge (X_train, all_data, left_index = True, right_index = True), чтобы объединить.Но все столбцы возвращаются путем добавления _x и _y к концу имени столбца для идентификации.

Мне просто нужны отдельные столбцы.

Спасибо!

Answer 1

Вы можете попытаться извлечь отдельные столбцы перед объединением, а затем явно передать их в команду объединения:

X_train_cols = X_train.columns
all_data_cols = all_data.columns
all_data_cols_new = list(set(all_data_cols).difference(X_train_cols))

Затем:

pd.merge(X_train, all_data[all_data_cols_new], left_index=True, right_index=True)