用戶定義的函數如式輸入中的R

Question

內置函數可以在式對象使用，例如

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

如何可以寫我的功能，使得它們可以在式對象使用？

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

Answer 1

你當然有什麼作品。一個問題是不同的建模函數以不同的方式處理公式。我認爲只要你返回model.matrix可以理解的東西，你會沒事的。那將意味着

1. 該函數被矢量化;即給定的長度爲N的向量，它返回一個結果也長度爲N

2. 它返回一個原子矢量或矩陣（但不是列表，或raw類型的）