假设我有一个数据矩阵,我想先将矩阵按行居中,然后按列计算均值。
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和处理它的维度。但这似乎有点尴尬。
只需告诉 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)