R函数计算偏差矩阵

Question

我编写了一个函数来计算一个矩阵，其中每一列是对应的输入矩阵列减去列均值。

# compute the deviation matrix
deviation <- function(X) {
  one <- rep(1, nrow(X))
  n <- ncol(X)
  d <- matrix(data = NA, nrow = nrow(X), ncol = ncol(X))
  for(i in seq.int(from = 1, to = n)) {
    d[,i] <- X[,i] - mean(X[,i], na.rm = TRUE) * one
  }
  d
}

这个函数可以用 R 更惯用地编写吗（也许使用函数式编程）？

Answer 1

使用sweep 和colMeans：

sweep(mat, 2, colMeans(mat))

默认情况下，sweep 使用 - 或减法函数，从每列 (MARGIN=2) 中的值中获取由 colMeans 计算的列均值。给出相同的结果：

mat <- matrix(1:12,nrow=3)
deviation(mat)
#     [,1] [,2] [,3] [,4]
#[1,]   -1   -1   -1   -1
#[2,]    0    0    0    0
#[3,]    1    1    1    1

sweep(mat, 2, colMeans(mat))
#     [,1] [,2] [,3] [,4]
#[1,]   -1   -1   -1   -1
#[2,]    0    0    0    0
#[3,]    1    1    1    1