计算矩阵中非零列的数量

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

Answer 2

我们可以使用apply按列计算具有any非零值的列数

sum(apply(mat != 0, 2, any))
#[1] 1

或者，如果您想要具有 all 非零值的列

sum(apply(mat != 0, 2, all))