Question

我有一个包含15000列的矩阵。有些单元格有float个值，有些单元格有0。我想完全删除所有值为0的所有列。

     col1     col2     col3     col4
row1  1        0        0        1
row2  3.4      0        0        2.4
row3  0.56     0        0        0
row4  0        0        0        0

我想删除列col2和col3并保留其余内容。我怎么能用R做呢？感谢

Answer 1

更快的方法是做同样的事情（快3到5倍）

M[,colSums(M^2) !=0]

编辑：添加了此处建议的各种方法的时间详细信息。 @Dwin使用M[, colSums(abs(M)) ! == 0]建议的方法似乎最快，特别是当矩阵很大时。如果建议采用其他解决方案，我会更新基准测试报告。

m <- cbind(rnorm(1000),0)
M <- matrix(rep(m,7500), ncol=15000)

f_joran   = function(M) M[, !apply(M==0,2,all)]
f_ramnath = function(M) M[, colSums(M^2) != 0]
f_ben     = function(M) M[, colSums(M==0) != ncol(M)]
f_dwin    = function(M) M[, colSums(abs(M)) != 0]

library(rbenchmark)
benchmark(f_joran(M), f_ramnath(M), f_ben(M), f_dwin(M), 
   columns = c('test', 'elapsed', 'relative'), 
   order = 'relative', replications = 10)


          test elapsed relative
4    f_dwin(M)  11.699 1.000000
2 f_ramnath(M)  12.056 1.030515
1   f_joran(M)  26.453 2.261133
3     f_ben(M)  28.981 2.477220

Answer 2

如何使用apply和all：

M <- as.matrix(data.frame(a=runif(10),b=rep(0,10),c=runif(10),d=rep(0,10)))

M[,which(!apply(M,2,FUN = function(x){all(x == 0)}))]

Answer 3

您提到了15,000列，但没有提到行数。如果有几千行，速度是一个问题，colSums将比apply快一点。

m <- cbind(rnorm(1000),0)
M <- matrix(rep(m,7500), ncol=15000)
system.time(foo <- M[,which(!apply(M==0,2,all))])
#   user  system elapsed 
#   1.63    0.23    1.86 
system.time(bar <- M[,colSums(M)!=0])
#   user  system elapsed 
#  0.340   0.060   0.413
identical(foo,bar)
# [1] TRUE

从矩阵中删除0的所有列

3 个答案: