在R中编写函数

Question

我的任务是在R中编写一个程序来定义函数y和值a = 1，b =2和e = normal distribution。然后该程序在x上回归y，并将系数提取为新变量。 到目前为止，我写下了：

Q3Data <- function(a = 1, b = 2, nPoints = 1000) {
x <- runif(nPoints);
y <- a + b * x + rnorm(nPoints, mean = 0, sd = 1)
  return(data.frame(x = x, y = y));
}

但我不确定如何继续。据我所知，我的意思是包含命令，以便在代码末尾的“}”之前执行所请求的任务，但我不确定如何继续。我知道如何将回归作为一个单独的步骤运行，而不是作为一个“连续”的代码串。

Answer 1

也许这就像你正在寻找的东西？它计算线性模型回归线并返回线系数。

Q3Data <- function(a = 1, b = 2, nPoints = 1000) {
    x <- runif(nPoints)
    y <- a + b * x + rnorm(nPoints, mean = 0, sd = 1)
    fit <- lm(y ~ x)
    fit$coef[2]
}

无需在R函数中使用return语句。从函数返回上次执行的命令的结果。