强制“应用”返回矩阵?

use*_*937 6 r

假设我有一个数据矩阵,我想先将矩阵按行居中,然后按列计算均值。

a=matrix(runif(50),nrow=5)
a1=apply(a,1,function(x)x-mean(x))
a.sum=apply(a1,1,sum)
Run Code Online (Sandbox Code Playgroud)

a有多个列时,这很有效。但是,有时输入只有一列,这会引起麻烦:

a=matrix(runif(5))
a1=apply(a,1,function(x)x-mean(x))
a.sum=apply(a1,1,sum)
Error in apply(a1, 1, sum) : dim(X) must have a positive length
Run Code Online (Sandbox Code Playgroud)

这是因为第一个apply返回了一个向量,而不是一个矩阵。R 自动删除了维度。那么有没有聪明的方法来防止这种情况发生?我知道我可以用不同的编码if来检测a和处理它的维度。但这似乎有点尴尬。

And*_*erg 1

只需告诉 R 它是一个矩阵:

a=matrix(runif(5))
#          [,1]
#[1,] 0.0103764
#[2,] 0.9738857
#[3,] 0.2845688
#[4,] 0.7050949
#[5,] 0.3000554

a1=as.matrix(apply(a,1,function(x)x-mean(x)))
#     [,1]
#[1,]    0
#[2,]    0
#[3,]    0
#[4,]    0
#[5,]    0

a.sum=apply(a1,1,sum)
#[1] 0 0 0 0 0
Run Code Online (Sandbox Code Playgroud)