如果rowSums大于1,则除以sum

时间:2017-09-07 22:12:08

标签: r

如果rowSums()大于1,我想除以行数之和。我没有想过没有for()循环的方法,所以我正在寻找一个没有循环的解决方案。

示例数据

dat <- structure(list(x1 = c(0.18, 0, 0.11, 0.24, 0.33), x2 = c(0.34, 
0.14, 0.36, 0.35, 0.21), x3 = c(0.1, 0.36, 0.12, 0.07, 0.18), 
    x4 = c(0.08, 0.35, 0.06, 0.1, 0.09), x5 = c(0.26, 0.13, 0.22, 
    0.31, 0.22)), .Names = c("x1", "x2", "x3", "x4", "x5"), row.names = c(NA, 
5L), class = "data.frame")

> rowSums(dat)

   1    2    3    4    5 
0.96 0.98 0.87 1.07 1.03 

我尝试了什么

这有效,但我想知道是否有更好的方法:

a <- which(rowSums(dat) > 1)
dat[a, ] <- dat[a,] / rowSums(dat[a,]

> rowSums(dat)
   1    2    3    4    5 
0.96 0.98 0.87 1.00 1.00 

2 个答案:

答案 0 :(得分:8)

这给出了与问题末尾附近的表达式相同的值:

dat / pmax(rowSums(dat), 1)

答案 1 :(得分:4)

这不如G.格洛腾迪克的回答,但您也可以使用ifelse

rs <- rowSums(dat)
dat / ifelse(rs < 1, rs, 1L)
         x1        x2        x3         x4        x5
1 0.1875000 0.3541667 0.1041667 0.08333333 0.2708333
2 0.0000000 0.1428571 0.3673469 0.35714286 0.1326531
3 0.1264368 0.4137931 0.1379310 0.06896552 0.2528736
4 0.2400000 0.3500000 0.0700000 0.10000000 0.3100000
5 0.3300000 0.2100000 0.1800000 0.09000000 0.2200000