Question

我有一个函数来计算二次方程的逆。默认情况下，它提供了两种可能的解决方案：

invquad<-function(a,b,c,y,roots="both")
{
    #Calculate the inverse of a quadratic function y=ax^2+bx+c (i.e. find x when given y.)
    #Gives NaN with non real solutions.
    root1<-sqrt((y-(c-b^2/(4*a)))/a)-(b/(2*a))
    root2<--sqrt((y-(c-b^2/(4*a)))/a)-(b/(2*a))
    if (roots=="both")
        result<-c(root1,root2)
    if (roots=="min")
        result<-min(root1,root2)
    if (roots=="max")
        result<-max(root1,root2)
    result
}

如果给出单个y值，这可以正常工作，但是如果我从数据框中给它一个列表或列，那么min和max元素给出了整个列表的最小值。我希望它只返回该元素的最小结果。我假设可以在列表上进行迭代，但效率不高。

有什么想法吗？

Answer 1

将min（max）函数替换为pmin（pmax）：

> invquad(1,2,3,3,"min")
[1] -2
> invquad(1,2,3,4,"min")
[1] -2.414214
> invquad(1,2,3,c(3,4),"min")
[1] -2.000000 -2.414214

Answer 2

查看plyr-package - 它允许将列表/数据框分成更小的部分，然后将您的函数应用于此子集。

如何分别对函数的每个结果应用Min或Max？

2 个答案: