在其他列的多个条件上添加具有行方式最大值的新列

时间:2013-05-31 00:03:26

标签: r

这应该相当容易,但我只是让自己感到困惑。使用数据框:

df <- data.frame(id = 1:5, rate1 = c(1, 0, 0.3, 0.5, 1)
, rate2 = c(0, 0.6, 1, 0, 1), cost1 = 3:7, cost2 = 8:4)

我想根据rate1rate2是否等于1添加新列,然后填写最新值cost1cost2的新列使用0.期望的结果将是:

  id rate1 rate2 cost1 cost2 overspent
1  1   1.0   0.0     3     8         8
2  2   0.0   0.6     4     7         0
3  3   0.3   1.0     5     6         6
4  4   0.5   0.0     6     5         0
5  5   1.0   1.0     7     4         7

我假设使用类似的东西:

df$overspent <- with(df, ifelse((rate1 == 1 || rate2 == 1)
                     , apply(cbind(cost1, cost2), 1, max), 0))

但显然它不起作用。请指正。谢谢!

1 个答案:

答案 0 :(得分:2)

您需要使用|,而不是||。但看看这有多优雅:

transform(df, overspent = ifelse(rate1 == 1 | rate2 == 1,
                                 pmax(cost1, cost2), 0))

另外,正如@ Ferdinand.kraft指出的那样,比较数字可能很危险(尽管不是在这个例子中),所以你可能想用rate1 == 1代替:{/ p>

abs(rate1 - 1) < .Machine$double.eps ^ 0.5

sapply(rate1 - 1, function(x)isTRUE(all.equal(x, 0)))