如何写个函数，根据输入的变量进行线性回归？

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

想写个这样的函数：

cal_lm <- function(x, y, dataset) {
    # 实现根据x、y定义的自变量因变量，求解线性回归结果
   
    # Args：
    # x : 自变量名
    # y：因变量名
    # data：数据集

    ans <- lm( y ~ x, data = dataset)
}

例如，数据集叫test，有5个变量，假设叫做aa, bb, cc, dd, ee，这个函数的功能就是，如果我想看bb ~ aa，就调用：

cal_lm(x = "aa", y = "bb", data = test)，则这个函数等同于lm(bb ~aa, data = test)

如果想跑bb~ aa + cc，则调用 cal_lm(x = c("aa", "cc"), y = "bb", data = test)

我试了试，关键是这个函数参数x，y的传递到lm中，似乎不能简单的用字符串来传递。请各位高人指点下~~谢谢！！

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：线性回归 如何写 如何

cal_lm <- function(x, y, dataset) {
  yy = dataset[,y]
  dataset = dataset[x]
  dataset$yy = yy
  ans <- lm(yy~., data = dataset)
  return(ans)
}