如何使用d3.js创建趋势/指数回归线?

时间:2012-11-27 18:42:33

标签: javascript math d3.js

我正在尝试创建一条曲线显示图表中的数据趋势,但我无法弄清楚如何生成必要的数据点,类似于此图像中的第二个图:

Least Squares Fitting Graphs

我发现的所有文档和示例都使用了数学,而且任何伪代码都会很棒。

1 个答案:

答案 0 :(得分:10)

我能够使用以下代码绘制指数回归线:

function square(x){return Math.pow(x,2);};

function array_sum(arr){
  var total = 0;
  arr.forEach(function(d){total+=d;});
  return total;
}

function exp_regression(Y){
  var n = Y.length;
  var X = d3.range(1,n+1);

  var sum_x = array_sum(X);
  var sum_y = array_sum(Y);
  var y_mean = array_sum(Y) / n;
  var log_y = Y.map(function(d){return Math.log(d)});
  var x_squared = X.map(function(d){return square(d)});
  var sum_x_squared = array_sum(x_squared);
  var sum_log_y = array_sum(log_y);
  var x_log_y = X.map(function(d,i){return d*log_y[i]});
  var sum_x_log_y = array_sum(x_log_y);

  a = (sum_log_y*sum_x_squared - sum_x*sum_x_log_y) /
      (n * sum_x_squared - square(sum_x));

  b = (n * sum_x_log_y - sum_x*sum_log_y) /
      (n * sum_x_squared - square(sum_x));

  var y_fit = [];
  X.forEach(function(x){
    y_fit.push(Math.exp(a)*Math.exp(b*x));
  });

  return y_fit;
}