如何在JavaScript中实现GROWTH函数

Question

我试图在JavaScript中实现Microsoft Excel的GROWTH函数。此函数通过使用现有数据来计算预测的指数增长。棘手的是它必须与多组known_x's值配合使用。我找不到任何参考方程式。有什么建议么？

在此先感谢您的帮助。

Answer 1

下面是基于my answer at math.SE重新实现从维基百科页面的simple linear regression公式：

function growth (known_y, known_x, new_x, use_const) { 
    // default values for optional parameters: 
    if (typeof(known_x) == 'undefined') { 
     known_x = []; 
     for (var i = 1; i <= known_y.length; i++) known_x.push(i); 
    } 
    if (typeof(new_x) == 'undefined') { 
     new_x = []; 
     for (var i = 1; i <= known_y.length; i++) new_x.push(i); 
    } 
    if (typeof(use_const) == 'undefined') use_const = true; 

    // calculate sums over the data: 
    var n = known_y.length; 
    var avg_x = 0; var avg_y = 0; var avg_xy = 0; var avg_xx = 0; 
    for (var i = 0; i < n; i++) { 
     var x = known_x[i]; var y = Math.log(known_y[i]); 
     avg_x += x; avg_y += y; avg_xy += x*y; avg_xx += x*x; 
    } 
    avg_x /= n; avg_y /= n; avg_xy /= n; avg_xx /= n; 

    // compute linear regression coefficients: 
    if (use_const) { 
     var beta = (avg_xy - avg_x*avg_y)/(avg_xx - avg_x*avg_x); 
     var alpha = avg_y - beta*avg_x; 
    } else { 
     var beta = avg_xy/avg_xx; 
     var alpha = 0; 
    } 
    // console.log("alpha = " + alpha + ", beta = " + beta); 

    // compute and return result array: 
    var new_y = []; 
    for (var i = 0; i < new_x.length; i++) { 
     new_y.push(Math.exp(alpha + beta * new_x[i])); 
    } 
    return new_y; 
} 

Here's a demo on ideone.com.您可以比较在Excel GROWTH documentation page演示表的输出。

请注意，算法中求和回路的数值稳定性可以通过使用诸如calculating the variance的维基百科页面上描述的技术来改进，如Kahan summation。然而，对于这样的简单例子，天真的求和循环相当不错。