RcppArmadillo 中的 fastLm 和 fastLmPure 函数之间的区别

Difference between fastLm and fastLmPure functions from RcppArmadillo

这是一个例子:

require(Rcpp)
require(RcppArmadillo)
require(zoo)
require(repmis)

myData <- source_DropboxData(file = "example.csv", 
                              key = "cbrmkkbssu5bn96", sep = ",", header = TRUE)

dolm = function(x) coef(fastLmPure(as.matrix(x[,2]), x[,1]))

myCoef = rollapply(myData, 260, dolm, by.column = FALSE)

summary(myCoef) # 80923 NA's

dolm2 = function(x) coef(fastLm(x[,1] ~ x[,2] + 0, data = as.data.frame(x)))

myCoef2 = rollapply(myData, 260, dolm2, by.column = FALSE)

summary(myCoef2) #  0 NA's 

在上面的示例中,使用 fastLmPure 的第一种方法在输出中产生 NA,而使用 fastLm 的第二种方法则不会。

这里是 link 到 fastLm & fastLmPure 函数写在 R:

https://github.com/RcppCore/RcppArmadillo/blob/master/R/fastLm.R

这是用 C++ 编写的 link 底层 fastLm 函数:

https://github.com/RcppCore/RcppArmadillo/blob/master/src/fastLm.cpp

根据这些 links 和 RcppArmadillo 的文档,我不明白是什么导致了输出的差异?为什么第二个输出中没有 NA?最重要的问题是什么例程/部分代码阻止 NA 出现在第二种方法中,它是如何实现的?

您正在使用两个不同的接口调用两个不同的函数

特别是,fastLm() 当通过 公式使用时 y ~ X 将依赖于 R 内部(和慢!!)函数来创建向量和对应于 fastLm(X, y).

的矩阵

这是一个简单的设置示例:

R> data(mtcars)
R> lm(mpg ~ cyl + disp + hp + wt - 1, data=mtcars)

Call:
lm(formula = mpg ~ cyl + disp + hp + wt - 1, data = mtcars)

Coefficients:
    cyl     disp       hp       wt  
 5.3560  -0.1206  -0.0313   5.6913  

R> fastLm(mpg ~ cyl + disp + hp + wt - 1, data=mtcars)

Call:
fastLm.formula(formula = mpg ~ cyl + disp + hp + wt - 1, data = mtcars)

Coefficients:
      cyl      disp        hp        wt 
 5.356014 -0.120609 -0.031306  5.691273 
R> fastLm(mtcars[, c("cyl","disp","hp","wt")], mtcars[,"mpg"])

Call:
fastLm.default(X = mtcars[, c("cyl", "disp", "hp", "wt")], y = mtcars[, 
    "mpg"])

Coefficients:
      cyl      disp        hp        wt 
 5.356014 -0.120609 -0.031306  5.691273 
R> 

现在让我们在左侧和右侧添加一个 NA。为了便于索引,我们将使用整行:

R> mtcars[7, ] <- NA
R> lm(mpg ~ cyl + disp + hp + wt - 1, data=mtcars)

Call:
lm(formula = mpg ~ cyl + disp + hp + wt - 1, data = mtcars)

Coefficients:
    cyl     disp       hp       wt  
 5.3501  -0.1215  -0.0332   5.8281  

R> fastLm(mpg ~ cyl + disp + hp + wt - 1, data=mtcars)

Call:
fastLm.formula(formula = mpg ~ cyl + disp + hp + wt - 1, data = mtcars)

Coefficients:
      cyl      disp        hp        wt 
 5.350102 -0.121478 -0.033184  5.828065 
R> fastLm(na.omit(mtcars[, c("cyl","disp","hp","wt")]), na.omit(mtcars[,"mpg"]))

Call:
fastLm.default(X = na.omit(mtcars[, c("cyl", "disp", "hp", "wt")]), 
    y = na.omit(mtcars[, "mpg"]))

Coefficients:
      cyl      disp        hp        wt 
 5.350102 -0.121478 -0.033184  5.828065 
R> 

更重要的是:结果在所有方法之间仍然相同,前提是我们对缺失值保持一致