r - 使用sapply()在两个参数上循环uniroot()

该函数根据给定的直径，流量和长度来计算管道中的压力损失。

hazwil2 <- function(diam, flow, leng){
  psi2=((1/(2.31*100))*1050*((flow/140)^1.852)*leng*diam^-4.87)
  return(psi2)
}

我正在寻找将压力损失保持在2 psi之内的最小直径。直径范围在2到12英寸之间。使用uniroot（）和两个任意值作为流和长度：

intercept   = 2L
uniroot(
  function(x) hazwil2(x, 100, 400) - intercept ,
  interval = c(2, 12)
)$root

现在，我尝试在流和长度值的这个数据集上循环uniroot

data <- read.csv(
  text =
  "leng,flow
  100, 100
  200, 100
  300, 100
  100, 150
  200, 150
  300, 150
  100, 200
  200, 200
  300, 200",
  stringsAsFactors = FALSE
)

最佳答案

考虑泛化f和l参数的unitroot调用，并在mapply中传递它（多次应用）以逐个元素向下迭代相等长度的向量（即列，数据帧的流和长度）：

find_root <- function(f, l) {
  intercept <- 2L
  uniroot(
    function(x) hazwil2(x, f, l) - intercept ,
    interval = c(2, 12)
  )$root
}

data$root <- mapply(find_root, data$flow, data$leng)

data
#   leng flow     root
# 1  100  100 2.681145
# 2  200  100 3.091243
# 3  300  100 3.359606
# 4  100  150 3.128141
# 5  200  150 3.606602
# 6  300  150 3.919727
# 7  100  200 3.489780
# 8  200  200 4.023564
# 9  300  200 4.372916

对于可能产生根本问题的较大集合，请考虑在tryCatch中包装函数调用以返回NA：

data <- expand.grid(leng = seq(100, 1000, by=100), flow = seq(10, 200, by=10))
data$root <- mapply(function(l,f) tryCatch(find_root(l,f), error=function(e) NA),
                    data$flow, data$leng)

输出量

head(data, 20)

#    leng flow     root
# 1   100   10       NA
# 2   200   10       NA
# 3   300   10       NA
# 4   400   10       NA
# 5   500   10       NA
# 6   600   10       NA
# 7   700   10       NA
# 8   800   10       NA
# 9   900   10       NA
# 10 1000   10       NA
# 11  100   20       NA
# 12  200   20       NA
# 13  300   20       NA
# 14  400   20       NA
# 15  500   20 2.023187
# 16  600   20 2.100367
# 17  700   20 2.167915
# 18  800   20 2.228178
# 19  900   20 2.282705
# 20 1000   20 2.332653

关于r - 使用sapply()在两个参数上循环uniroot()，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/50978973/

Uniroot

r - 使用sapply()在两个参数上循环uniroot()