钳位变量在

Question

给定两个值x和b，我想要一个函数来将x限制在[-b, b]范围内。特别是：

• 如果x小于或等于-b，则该函数返回-b;
• 如果x大于-b且小于b，则该函数返回x;
• 如果x大于或等于b，则该函数返回b。

在R中我编写了以下函数truncfn。只有部分功能有效。我哪里弄错了？有更简单的方法吗？

b <- 5

truncfn <- function(x){
  if((x<(-b))||(x==-b)) -b 
  if((x>(-b))&&(x<b)) x 
  if((x>b)||(x==b)) b
}

truncfn(10)
5
truncfn(4)
truncfn(-10)

Answer 1

我会使用pmin和pmax函数来执行此操作，这是一个没有if语句的单个表达式：

b <- 5
x <- seq(-10, 10)
pmax(pmin(x, b), -b)
# [1] -5 -5 -5 -5 -5 -5 -4 -3 -2 -1  0  1  2  3  4  5  5  5  5  5  5

您可以使用if语句替换第二个和第三个else if语句（或者使用return(b)，return(x)和return(-b)）来修复您的功能。这是一个工作版本（我冒昧地使用<=而不是单独检查<和==）：

truncfn <- function(x){
  if (x <= -b) -b
  else if (x > -b && x < b) x 
  else b
}
truncfn(10)
# [1] 5
truncfn(4)
# [1] 4
truncfn(-10)
# [1] -5

但请注意，我使用pmin和pmax编写的函数可以将整个向量作为输入，而您的只能使用单个数字。

Answer 2

您可以使用raster::clamp

library(raster)
b <- 5
v <- -10:10
raster::clamp(v, -b, b)
# [1] -5 -5 -5 -5 -5 -5 -4 -3 -2 -1  0  1  2  3  4  5  5  5  5  5  5