总和大于r

时间:2015-05-06 14:45:44

标签: r sumifs

我有一个包含145行和1000多列的数据帧(obs)以及一个包含145个值(thr)的数字向量。

我想得到另一个含有145个元素的向量(sumifs),其中每个元素是obs [n,]> = thr [n]的值的总和。

我以为我可以运行一个for循环,其中单行和计算或多或少像:

sumifs[n] <- if(obs[n,]>=thr[n],sum(obs[n,]))

但我没有设法使其适用于单行。

我一直在查看有关建议使用聚合或plyr包的其他问题,但我没有找到任何结果。

是一个只有15行和3列的简化示例
c1 <- rep(1:5,3)
c2 <- rep(3:7,3)
c3 <- rep(2:6,3)

obs <- data.frame(r1,r2,r3)
thr <- c(2,2,3,3,4,4,5,5,2,2,3,3,4,4,5)

obs
   r1 r2 r3
1   1  3  2
2   2  4  3
3   3  5  4
4   4  6  5
5   5  7  6
6   1  3  2
7   2  4  3
8   3  5  4
9   4  6  5
10  5  7  6
11  1  3  2
12  2  4  3
13  3  5  4
14  4  6  5
15  5  7  6

因此,sumif应该是:

sumifs
5
9
12
15
18
0
0
0
15
18
3
7
9
15
18

1 个答案:

答案 0 :(得分:5)

#your data
DF <- as.data.frame(matrix(1:6, ncol = 2))
#turn into matrix
m <- as.matrix(DF)

#your threshold
thr <- c(3, 1, 7)

#compare
m >= thr
#        V1    V2
#[1,] FALSE  TRUE
#[2,]  TRUE  TRUE
#[3,] FALSE FALSE

#logical values get turned to 0/1 during arithmetics
#thus we can just multiply the matrix with the comparison
m * (m >= thr)
#     V1 V2
#[1,]  0  4
#[2,]  2  5
#[3,]  0  0

#and calculate the row sums
rowSums(m * (m >= thr))
#[1] 4 7 0