Question

我有一个类似于以下数据表：

| ID | C1 | C2 | C3  |
|  1 |  1 | 20 | 400 |
|  2 |  0 | 15 | 500 |
|  3 |  0 |  2 | 350 |
|  4 |  1 |  4 | 402 |
|  5 |  1 |  8 | 333 |

我想使用这样的函数生成一个新列“ CALC”：

func <- function(a, b, c){
      pol <- a*b-0.01*c
      value <- min(max(pol, 0), 5)
      }

我认为这将给我每行pol解决方案，并以0和5为上限。所以，代替了

| ID | C1 | C2 | C3  | CALC  |
|  1 |  1 | 20 | 400 | 16    |
|  2 |  0 | 15 | 500 | -5    |
|  3 |  0 |  2 | 350 | -3.5  |
|  4 |  1 |  4 | 402 | -0.02 |
|  5 |  1 |  8 | 333 | 4.67  |

它应该返回

| ID | C1 | C2 | C3  | CALC  |
|  1 |  1 | 20 | 400 |  5    |
|  2 |  0 | 15 | 500 |  0    |
|  3 |  0 |  2 | 350 |  0    |
|  4 |  1 |  4 | 402 |  0    |
|  5 |  1 |  8 | 333 | 4.67  |

但是，它提供的是以下dt：

| ID | C1 | C2 | C3  | CALC  |
|  1 |  1 | 20 | 400 |  5    |
|  2 |  0 | 15 | 500 |  5    |
|  3 |  0 |  2 | 350 |  5    |
|  4 |  1 |  4 | 402 |  5    |
|  5 |  1 |  8 | 333 |  5    |

因此，如果我没有记错的话，它将获取列中的最大值（将其上限为5）并沿列放置。不是我的意图，那将是单独获得每一行的最大值。

是否有任何“简便”的方法可以解决？原始dt更复杂，因此不可能简单地进行dt[, CALC := func(C1, C2, C3), by="ID"]，因为它应该像by=.(GROUP1, GROUP2, ID)之类，而且我敢肯定它必须更简单。也许有一种简单的方法可以遍历函数本身？

非常感谢您的时间。

Answer 1

您需要使用pmin和pmax

func <- function(a, b, c){
  pol <- a*b-0.01*c
  pmin(pmax(pol, 0), 5)
}


transform(dat,s=func(C1,C2,C3))
  ID C1 C2  C3    s
1  1  1 20 400 5.00
2  2  0 15 500 0.00
3  3  0  2 350 0.00
4  4  1  4 402 0.00
5  5  1  8 333 4.67

 setDT(dat)[,CALC:=func(C1,C2,C3)]
 dat
   ID C1 C2  C3 CALC
1:  1  1 20 400 5.00
2:  2  0 15 500 0.00
3:  3  0  2 350 0.00
4:  4  1  4 402 0.00
5:  5  1  8 333 4.67

在data.table中生成具有上限值的列

1 个答案: