进行多元线性回归,我想删除一个变量,但不知道如何将正确的语句写入corr = df.corr()
。
在使用pearson进行关联之前,我已使用
df = df.drop('location', axis=1)
但是也许有更聪明的方法吗?
有人知道吗?
答案 0 :(得分:0)
通常,您的操作是正确的。如果您希望将location
变量保留在原始df
中,然后将其从关联表中删除,则可以在一行中编写代码:
corr = df.drop('location', axis = 1).corr()
由于df.drop('location', axis = 1)
返回一个数据框,因此您可以简单地应用数据框类中包含的任何函数,例如df.corr()
。