d3-regression

# d3-regression Calculate statistical regressions from two-dimensional data. [<img alt="Stastical Regressions" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/cover.png" width="882">](https://observablehq.com/@harrystevens/introducing-d3-regression) ## Installing If you use NPM, `npm install d3-regression`. Otherwise, download the [latest release](https://github.com/HarryStevens/d3-regression/raw/master/dist/d3-regression.zip). AMD, CommonJS, and vanilla environments are supported. In vanilla, a d3 global is exported: ```html <script src="https://unpkg.com/d3-regression@1.3.10/dist/d3-regression.min.js"></script> <script> const regression = d3.regressionLinear() .x(d => d.x) .y(d => d.y) .domain([0, 100]); </script> ``` ## API Reference - [Linear](#regressionLinear) - [Exponential](#regressionExp) - [Logarithmic](#regressionLog) - [Quadratic](#regressionQuad) - [Polynomial](#regressionPoly) - [Power law](#regressionPow) - [LOESS](#regressionLoess) <a name="regressionLinear" href="#regressionLinear">#</a> d3.regressionLinear() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/linear.js "Source"), [Example](https://observablehq.com/@harrystevens/linear-regression "Example") Creates a new linear regression generator with default [x-](#linear_x) and [y-](#linear_y) accessors and a null [domain](#linear_domain). [<img alt="Linear regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/linear.png" width="250">](https://observablehq.com/@harrystevens/linear-regression) <a name="_linear" href="#_linear">#</a> linear(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/linear.js#L9 "Source") Computes the linear regression, which takes the form y = ax + b, for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a line represented as an array of two points, where each point is an array of two numbers representing the point's coordinates. Also returns properties a and b, representing the equation's coefficients, and rSquared, representing the coefficient of determination. Lastly, returns a predict property, which is a function that outputs a y-coordinate given an input x-coordinate. <a name="linear_x" href="#linear_x">#</a> linear.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/linear.js#L47 "Source") If x is specified, sets the x-coordinate accessor, which is passed passed the current datum (d), the current index (i), and the entire data array (data). If x is not specified, returns the current x-coordinate accessor, which defaults to: ```js function x(d, i, data) { return d[0]; } ``` <a name="linear_y" href="#linear_y">#</a> linear.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/linear.js#L51 "Source") If y is specified, sets the y-coordinate accessor, which is passed passed the current datum (d), the current index (i), and the entire data array (data). If y is not specified, returns the current y-coordinate accessor, which defaults to: ```js function y(d, i, data) { return d[1]; } ``` <a name="linear_domain" href="#linear_domain">#</a> linear.domain([domain]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/linear.js#L43 "Source") If domain is specified, sets the minimum and maximum x-coordinates of the returned line to the specified array of numbers. The array must contain two elements. If the elements in the given array are not numbers, they will be coerced to numbers. If domain is not specified, returns a copy of the regression generator’s current domain. If data is passed to the regression generator before a domain has been specified, the domain will be set to the minimum and maximum x-coordinate values of the data. <a name="regressionExp" href="#regressionExp">#</a> d3.regressionExp() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/exponential.js "Source"), [Example](https://observablehq.com/@harrystevens/exponential-regression "Example") Creates a new exponential regression generator with default [x-](#exp_x) and [y-](#exp_y) accessors and a null [domain](#exp_domain). [<img alt="Exponential regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/exponential-2.png" width="250">](https://observablehq.com/@harrystevens/exponential-regression) <a name="_exponential" href="#_exponential">#</a> exp(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/exponential.js#L10 "Source") Computes the exponential regression, which takes the form y = aebx, for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a smooth line represented as an array of points, where each point is an array of two numbers representing the point's coordinates. Also returns properties a and b, representing the equation's coefficients, and rSquared, representing the coefficient of determination. Lastly, returns a predict property, which is a function that outputs a y-coordinate given an input x-coordinate. <a name="exp_x" href="#exp_x">#</a> exp.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/exponential.js#L53 "Source") See [linear.x()](#linear_x). <a name="exp_y" href="#exp_y">#</a> exp.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/exponential.js#L57 "Source") See [linear.y()](#linear_y). <a name="exp_domain" href="#exp_domain">#</a> exp.domain([domain]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/exponential.js#L49 "Source") See [linear.domain()](#linear_domain). <a name="regressionLog" href="#regressionLog">#</a> d3.regressionLog() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/logarithmic.js "Source"), [Example](https://observablehq.com/@harrystevens/logarithmic-regression "Example") Creates a new logarithmic regression generator with default [x-](#log_x) and [y-](#log_y) accessors and a null [domain](#log_domain). [<img alt="Logarithmic regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/logarithmic.png" width="250">](https://observablehq.com/@harrystevens/logarithmic-regression) <a name="_log" href="#_log">#</a> log(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/logarithmic.js#L12 "Source") Computes the logarithmic regression, which takes the form y = a · ln(x) + b, for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a smooth line represented as an array of points, where each point is an array of two numbers representing the point's coordinates. Also returns properties a and b, representing the equation's coefficients, and rSquared, representing the coefficient of determination. Lastly, returns a predict property, which is a function that outputs a y-coordinate given an input x-coordinate. <a name="log_x" href="#log_x">#</a> log.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/logarithmic.js#L52 "Source") See [linear.x()](#linear_x). <a name="log_y" href="#log_y">#</a> log.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/logarithmic.js#L56 "Source") See [linear.y()](#linear_y). <a name="log_domain" href="#log_domain">#</a> log.domain([domain]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/logarithmic.js#L48 "Source") See [linear.domain()](#linear_domain). <a name="log_base" href="#log_base">#</a> log.base([base]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/logarithmic.js#L60 "Source") If base is specified, sets the base of the logarithmic regression. If base is not specified, returns the current base, which defaults to [Euler's number](https://en.wikipedia.org/wiki/E_(mathematical_constant)). <a name="regressionQuad" href="#regressionQuad">#</a> d3.regressionQuad() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/quadratic.js "Source"), [Example](https://observablehq.com/@harrystevens/quadratic-regression "Example") Creates a new quadratic regression generator with default [x-](#quad_x) and [y-](#quad_y) accessors and a null [domain](#quad_domain). [<img alt="Quadratic regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/quadratic.png" width="250">](https://observablehq.com/@harrystevens/quadratic-regression) <a name="_quad" href="#_quad">#</a> quad(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/quadratic.js#L9 "Source") Computes the quadratic regression, which takes the form y = ax2 + bx + c, for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a smooth line represented as an array of points, where each point is an array of two numbers representing the point's coordinates. Also returns properties a, b, and <e>c</e>, representing the equation's coefficients, and rSquared, representing the coefficient of determination. Lastly, returns a predict property, which is a function that outputs a y-coordinate given an input x-coordinate. <a name="quad_x" href="#quad_x">#</a> quad.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/quadratic.js#L71 "Source") See [linear.x()](#linear_x). <a name="quad_y" href="#quad_y">#</a> quad.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/quadratic.js#L75 "Source") See [linear.y()](#linear_y). <a name="quad_domain" href="#quad_domain">#</a> quad.domain([domain]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/quadratic.js#L67 "Source") See [linear.domain()](#linear_domain). <a name="regressionPoly" href="#regressionPoly">#</a> d3.regressionPoly() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/polynomial.js "Source"), [Example](https://observablehq.com/@harrystevens/polynomial-regression "Example") Creates a new polynomial regression generator with default [x-](#poly_x) and [y-](#poly_y) accessors, a null [domain](#poly_domain), and an [order](#poly_order) of 3. This implementation was adapted from [regression-js](https://github.com/Tom-Alexander/regression-js/). [<img alt="Polynomial regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/polynomial.png" width="250">](https://observablehq.com/@harrystevens/polynomial-regression) <a name="_poly" href="#_poly">#</a> poly(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/polynomial.js#L13 "Source") Computes the polynomial regression, which takes the form y = anxn + ... + a1x + a0, for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a smooth line represented as an array of points, where each point is an array of two numbers representing the point's coordinates. Also returns three properties: coefficients, an array representing the equation's coefficients with the intercept as the first item and nth degree coefficient as the last item; rSquared, representing the coefficient of determination; and predict, a function that outputs a y-coordinate given an input x-coordinate. <a name="poly_x" href="#poly_x">#</a> poly.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/polynomial.js#L92 "Source") See [linear.x()](#linear_x). <a name="poly_y" href="#poly_y">#</a> poly.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/polynomial.js#L96 "Source") See [linear.y()](#linear_y). <a name="poly_domain" href="#poly_domain">#</a> poly.domain([domain]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/polynomial.js#L88 "Source") See [linear.domain()](#linear_domain). <a name="poly_order" href="#poly_order">#</a> poly.order([order]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/polynomial.js#L100 "Source") If order is specified, sets the regression's order to the specified number. For example, if order is set to 4, the regression generator will perform a fourth-degree polynomial regression. Likewise, if order is set to 2, the regression generator will perform a quadratic regression. Be careful about attempting to fit your data with higher order polynomials; though the regression line will fit your data with a high determination coefficient, it may have little predictive power for data outside of your domain. If order is not specified, returns the regression generator's current order, which defaults to 3. <a name="regressionPow" href="#regressionPow">#</a> d3.regressionPow() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/power.js "Source"), [Example](https://observablehq.com/@harrystevens/power-law-regression "Example") Creates a new power law regression generator with default [x-](#pow_x) and [y-](#pow_y) accessors and a null [domain](#pow_domain). [<img alt="Power law regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/power.png" width="250">](https://observablehq.com/@harrystevens/power-law-regression) <a name="_pow" href="#_pow">#</a> pow(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/power.js#L10 "Source") Computes the power law regression, which takes the form y = axb, for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a smooth line represented as an array of points, where each point is an array of two numbers representing the point's coordinates. Also returns properties a and b, representing the equation's coefficients, and rSquared, representing the coefficient of determination. Lastly, returns a predict property, which is a function that outputs a y-coordinate given an input x-coordinate. <a name="pow_x" href="#pow_x">#</a> pow.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/power.js#L54 "Source") See [linear.x()](#linear_x). <a name="pow_y" href="#pow_y">#</a> pow.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/power.js#L58 "Source") See [linear.y()](#linear_y). <a name="pow_domain" href="#pow_domain">#</a> pow.domain([domain]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/power.js#L50 "Source") See [linear.domain()](#linear_domain). <a name="regressionLoess" href="#regressionLoess">#</a> d3.regressionLoess() · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/loess.js "Source"), [Example](https://observablehq.com/@harrystevens/loess-regression "Example") Creates a new [LOESS regression](https://en.wikipedia.org/wiki/Local_regression) generator with default [x-](#loess_x) and [y-](#loess_y) accessors and a [bandwidth](#loess_bandwidth) of .3. This implementation was adapted from [science.js](https://github.com/jasondavies/science.js). [<img alt="LOESS regression" src="https://raw.githubusercontent.com/harrystevens/d3-regression/master/img/loess.png" width="250">](https://observablehq.com/@harrystevens/loess-regression) <a name="_loess" href="#_loess">#</a> loess(data) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/loess.js#L14 "Source") Computes the LOESS regression for the specified *data* points, ignoring points with invalid values (null, undefined, NaN, Infinity). Returns a line represented as an array of n points, where each point is an array of two numbers representing the point's coordinates. <a name="loess_x" href="#loess_x">#</a> loess.x([x]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/loess.js#L81 "Source") See [linear.x()](#linear_x). <a name="loess_y" href="#loess_y">#</a> loess.y([y]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/loess.js#L85 "Source") See [linear.y()](#linear_y). <a name="loess_bandwidth" href="#loess_bandwidth">#</a> loess.bandwidth([bandwidth]) · [Source](https://github.com/harrystevens/d3-regression/blob/master/src/loess.js#L77 "Source") If bandwidth is specified, sets the LOESS regression's bandwidth, or smoothing parameter, to the specific number between 0 and 1. The bandwidth represents the share of the total data points that are used to calculate each local fit. Higher bandwidths produce smoother lines, and vice versa. If bandwidth is not specified, returns a copy of the regression generator’s current bandwidth, which defaults to .3.