用户定义函数作为公式输入

Question

R中的内置函数可以在公式对象中使用，例如

reg1 = lm(y ~ log(x), data = data1)

如何编写函数以便它们可以在公式对象中使用？

fnMyFun = function(x) {
  return(x^2)
}
reg2  = lm(y ~ fnMyFun(x), data = data1)

- tchakravarty

Answer 1

您现有的内容肯定是有效的。问题之一是不同的建模函数以不同的方式处理公式。我认为只要您返回一个model.matrix可以理解的东西，您就会没问题。这意味着：

1. 该函数需要矢量化；即给定长度为N的向量，它也返回长度为N的结果。
2. 它必须返回原子向量或矩阵（但不是列表或类型raw）。

- Hong Ooi