使用corr = df.corr()删除变量

时间:2019-01-13 12:40:12

标签: python pandas dataframe

进行多元线性回归,我想删除一个变量,但不知道如何将正确的语句写入corr = df.corr()

在使用pearson进行关联之前,我已使用

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

但是也许有更聪明的方法吗?

有人知道吗?

1 个答案:

答案 0 :(得分:0)

通常,您的操作是正确的。如果您希望将location变量保留在原始df中,然后将其从关联表中删除,则可以在一行中编写代码:

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

由于df.drop('location', axis = 1)返回一个数据框,因此您可以简单地应用数据框类中包含的任何函数,例如df.corr()