Blame view

node_modules/echarts/extension-src/dataTool/prepareBoxplotData.ts 3.48 KB
bd028579   易尊强   2/28
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
  /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership.  The ASF licenses this file
  * to you under the Apache License, Version 2.0 (the
  * "License"); you may not use this file except in compliance
  * with the License.  You may obtain a copy of the License at
  *
  *   http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing,
  * software distributed under the License is distributed on an
  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  * KIND, either express or implied.  See the License for the
  * specific language governing permissions and limitations
  * under the License.
  */
  
  
  function asc<T extends number[]>(arr: T): T {
      arr.sort(function (a, b) {
          return a - b;
      });
      return arr;
  }
  
  function quantile(ascArr: number[], p: number): number {
      const H = (ascArr.length - 1) * p + 1;
      const h = Math.floor(H);
      const v = +ascArr[h - 1];
      const e = H - h;
      return e ? v + e * (ascArr[h] - v) : v;
  }
  /**
   * See:
   *  <https://en.wikipedia.org/wiki/Box_plot#cite_note-frigge_hoaglin_iglewicz-2>
   *  <http://stat.ethz.ch/R-manual/R-devel/library/grDevices/html/boxplot.stats.html>
   *
   * Helper method for preparing data.
   *
   * @param {Array.<number>} rawData like
   *        [
   *            [12,232,443], (raw data set for the first box)
   *            [3843,5545,1232], (raw data set for the second box)
   *            ...
   *        ]
   * @param {Object} [opt]
   *
   * @param {(number|string)} [opt.boundIQR=1.5] Data less than min bound is outlier.
   *      default 1.5, means Q1 - 1.5 * (Q3 - Q1).
   *      If 'none'/0 passed, min bound will not be used.
   * @param {(number|string)} [opt.layout='horizontal']
   *      Box plot layout, can be 'horizontal' or 'vertical'
   * @return {Object} {
   *      boxData: Array.<Array.<number>>
   *      outliers: Array.<Array.<number>>
   *      axisData: Array.<string>
   * }
   */
  export default function (
      rawData: number[][],
      opt: {
          boundIQR?: number | 'none',
          layout?: 'horizontal' | 'vertical'
      }
  ): {
      boxData: number[][]
      outliers: number[][]
      axisData: string[]
  } {
      opt = opt || {};
      const boxData = [];
      const outliers = [];
      const axisData: string[] = [];
      const boundIQR = opt.boundIQR;
      const useExtreme = boundIQR === 'none' || boundIQR === 0;
  
      for (let i = 0; i < rawData.length; i++) {
          axisData.push(i + '');
          const ascList = asc(rawData[i].slice());
  
          const Q1 = quantile(ascList, 0.25);
          const Q2 = quantile(ascList, 0.5);
          const Q3 = quantile(ascList, 0.75);
          const min = ascList[0];
          const max = ascList[ascList.length - 1];
  
          const bound = (boundIQR == null ? 1.5 : boundIQR as number) * (Q3 - Q1);
  
          const low = useExtreme
              ? min
              : Math.max(min, Q1 - bound);
          const high = useExtreme
              ? max
              : Math.min(max, Q3 + bound);
  
          boxData.push([low, Q1, Q2, Q3, high]);
  
          for (let j = 0; j < ascList.length; j++) {
              const dataItem = ascList[j];
              if (dataItem < low || dataItem > high) {
                  const outlier = [i, dataItem];
                  opt.layout === 'vertical' && outlier.reverse();
                  outliers.push(outlier);
              }
          }
      }
      return {
          boxData: boxData,
          outliers: outliers,
          axisData: axisData
      };
  }