用 R 中 m*n 数值矩阵的每一列的正元素之和创建一个向量

Question

我需要创建一个 R 函数，它以任意格式 n*m 的数值矩阵 A 作为输入，并返回一个向量，该向量与 A 的列数一样长，其中包含每列正元素的总和。

我必须以两种方式做到这一点——第一种在嵌套循环中，第二种作为使用向量/矩阵运算的单列。

到目前为止，我已经想出了以下代码，它创建了矩阵 A 的列数量大小的向量，但我似乎只能让它给我矩阵的所有正元素的总和，而不是每个专栏：

colSumPos(A){

columns <- ncol(A)

v1 <- vector("numeric", columns)

    for(i in 1:columns)
    {
        v1[i] <- sum(A[which(A>0)])
    }
}

有人能解释一下我如何分别得到每列的总和，然后我可以如何简化代码来处理嵌套循环吗？

提前感谢您的帮助！

Answer 1

我们可以使用apply 和MARGIN=2 循环遍历列并获取大于0 的元素的sum

apply(A, 2, function(x) sum(x[x >0], na.rm = TRUE))
#[1] 1.8036685 0.7129192 0.9305136 2.6625824 0.0000000

或者另一个选项是colSums 将小于或等于 0 的值替换为 NA

colSums(A*NA^(A<=0), na.rm = TRUE)
#[1] 1.8036685 0.7129192 0.9305136 2.6625824 0.0000000

或者通过更直接的方法

colSums(replace(A, A<=0, NA), na.rm = TRUE)
#[1] 1.8036685 0.7129192 0.9305136 2.6625824 0.0000000

或者如果没有 NA 元素（不需要na.rm=TRUE），我们可以将小于或等于 0 的值替换为 0 并使其紧凑（如 @ikop 评论的那样）

colSums(A*(A>0))
#[1] 1.8036685 0.7129192 0.9305136 2.6625824 0.0000000

数据

set.seed(24)
A <- matrix(rnorm(25), 5, 5)

Answer 2

如果你使用 for 循环，你可以尝试下面的代码

sumColum <- function(A){
for(i in 1:nrow(A)){
    for(j in 1:ncol(A)){
        colSums(replace(A, A<=0, NA), na.rm = TRUE)
    }
}
colSums(A)

}