函数输出为列表时使用 R 函数曲线

Question

当函数的输出是一个简单的表达式（例如单个数字）时，我熟悉使用函数curve 在 R 中绘制函数。我们如何处理输出有点复杂的函数，比如列表？

一个让我自己清楚的例子。假设我有以下功能。

f <- function(a,b,c,d){
    check <- a
    prob <- c:d
    n <- c*2
    r <- d-b
    lista <- list(check,prob,n,r)
    names(lista) <- c("check", "prob", "n", "r")
    return(lista)
}

f <- Vectorize(f, vectorize.args = "a")

如何使用函数curve 来绘制输出列表中的一个元素的结果，比如n，作为a 的函数？如果我将返回设置为 onlyn，这将非常顺利，但如果我想保留当前列表输出用于其他目的怎么办？

我检查了?curve，它指出，相当令人沮丧的是，

对于计算成本高昂的表达式，您应该使用更智能的工具

不要认为这是适用的情况，但实际上，正如帮助页面所建议的那样，“方式curvehandlesexpr已经引起了混乱”。

一种天真的方法是调用

curve(f(x, b= 2, c= 3, d=1)$n)

但这似乎并没有起到作用。无需更改函数输出或不必使用“ggplot”和排序的任何解决方案？

Answer 1

您应该在矢量化之前提取元素。也就是说，

f <- function(a,b,c,d){
    check <- a
    prob <- c:d
    n <- c*2
    r <- d-b
    lista <- list(check,prob,n,r)
    names(lista) <- c("check", "prob", "n", "r")
    return(lista)
}
fn <- Vectorize(function(x) f(x, b= 2, c= 3, d=1)$n)
curve(fn)

如果你真的想使用f的原始向量化版本，那么你需要意识到结果取决于你的函数的结果，而不是简单的数据框或列表。在您的情况下，它是一个有 4 行的列表模式矩阵。所以你可以使用

f <- function(a,b,c,d){
    check <- a
    prob <- c:d
    n <- c*2
    r <- d-b
    lista <- list(check,prob,n,r)
    names(lista) <- c("check", "prob", "n", "r")
    return(lista)
}

f <- Vectorize(f, vectorize.args = "a")

curve((function(a) unlist(f(a, b=2, c=3, d=1)["n",]))(x))

或

fn <- function(a) unlist(f(a, b=2, c=3, d=1)["n",])

curve(fn)

（可能还有其他可以使用的表达式），但真的值得弄清楚这一点吗？