我想使用 R 定义一个分段函数,但是,我的 R 代码出错了。欢迎任何建议。
x<-seq(-5, 5, by=0.01)
for (x in -5:5){
if (-0.326 < x < 0.652) fx<- 0.632
else if (-1.793<x<-1.304) fx<- 0.454
else if (1.630<x<2.119) fx<-0.227
else fx<- 0 }
试试这个:
x <- seq(-5, 5, 0.01)
fx <- (x > -0.326 & x <0.625) * 0.632 +
(x > -1.793 & x < -1.304) * 0.454 +
(x > 1.630 & x < 2.119) * 0.227
plot(x, fx)
我参加聚会有点晚了,但我忍不住发布了更多的方法来做到这一点。两者都利用 R 功能来处理实线上的间隔。
如果您在向量中定义切点和函数值,cuts
如下vals
所示:
cuts <- c( -Inf, -1.793, -1.304, -0.326, 0.625, 1.630, 2.119 )
vals <- c( 0, 0.454, 0, 0.632, 0, 0.227, 0 )
然后,您可以使用findInterval
有效地查找x
切点中的值:
fx <- vals[findInterval(x, c(-Inf, cuts))]
如果这个函数需要做一些比查找常量值更高级的事情,你可以将表达式或函数或任何你想要的东西放在 中,如果你愿意vals
,可以使用 a list
。
或者,由于此函数是一个阶跃函数,您可以使用stepfun
:
f <- stepfun(cuts[-1], vals)
fx <- f(x)
然后你也可以使用漂亮的绘图方法stepfun
。
或者你可以使用ifelse
.
fx <- ifelse(x > -0.326 & x <0.625, 0.632,
ifelse(x > -1.793 & x < -1.304, 0.454,
ifelse(x > 1.630 & x < 2.119, 0.227, 0)))
也许如果你拆分条件
if((-1.793<x) & (x < 0.652)) ...
编辑:这似乎还不是全部,这是一种不同的方法:
x<-seq(-5, 5, by=0.01)
fx <- function(x) {
res <- rep(0, length(x))
res[(-0.326 < x) & (x < 0.652)] <- 0.632
res[(-1.793<x) & (x < (-1.304))] <- 0.454
res[(1.630<x) & (x <2.119)] <- 0.227
return(res)
}
fx(x)
还有另一种选择,这次使用cut
.
regions <- c(-Inf, -1.793, -1.304, -0.326, 0.652, 1.63, 2.119, Inf)
group <- cut(x, regions)
f_values <- c(0, 0.454, 0, 0.632, 0, 0.227, 0)
(fx <- f_values[group])
除非您有不同的截止点,否则我会使用switch
. 这是一个带有简化切割值的示例。
xcuts<-1:10 #the values at which you change fx assignment
xx<- seq(1.5,10,5, by =10) #vector of fx values to be selected
switch(max(which(x>xcuts)),
1= fx<-xx[1],
2= fx<-xx[2],
..."et cetera"...
)
循环 x。