r - 计算矩阵中非零列的数量

Question

我有一个矩阵mat。 mat 的某些列全部为零，而某些列不为零。如何计算非零列？

mat<-matrix(rep(0,2*5),ncol=2)
mat[,1]=c(1,2,3,4,5)

期望的结果是 1。

Answer 1

如果您想要具有非零元素的列，这意味着它们可能具有零但并非所有行都是零，那么以下工作:

mat<-matrix(rep(0,2*5),ncol=2)
mat[,1]=c(1,2,3,4,5)

sum(colSums(mat!=0)!=0)
#> [1] 1

但是如果你想找到没有零的列，请看下面。我正在举一个例子来更好地说明这一点:

mat2<-matrix(rep(0,4*5),ncol=4)
mat2[,1]=c(0,2,3,4,5)
mat2[,2]=c(1,2,3,4,5)
mat2[,3]=c(0,0,0,-1,1)

sum(colSums(mat2!=0)!=0) #count of non-zero columns
#> [1] 3

sum(colSums(mat2!=0)==nrow(mat2)) #count of columns with no zeros
#> [1] 1

如果您想找出哪些列不为零，请使用 which 而不是 sum:

which(colSums(mat2!=0)!=0) #non-zero columns
#> [1] 1 2 3


which(colSums(mat2!=0)==nrow(mat2)) #columns with no zeros
#> [1] 2