Удалить переменную используя corr = df.corr ()

Question

Делая многолинейную регрессию, я хотел бы отбросить одну переменную, но не знаю, как записать правильное утверждение в corr = df.corr().

Перед использованием Пирсона для корреляции я использовал

df = df.drop('location', axis=1)

но, может быть, есть более разумный способ сделать это?

Кто-нибудь знает как?

Answer 1

В целом, ваша процедура верна. Если вы предпочитаете оставить переменную location в исходном df и просто удалить ее из таблицы корреляции, вы можете закодировать ее в одну строку:

corr = df.drop('location', axis = 1).corr()

Поскольку df.drop('location', axis = 1) возвращает фрейм данных, вы можете просто применить любую функцию, включенную в класс фреймов данных, например df.corr().