@antv/g2plot
Version:
An interactive and responsive charting library
82 lines • 3.23 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.binHistogram = void 0;
var util_1 = require("@antv/util");
// 进行转换得到值所在的 range
function getBinKey(value, binWidth, binNumber) {
// 做一点特殊处理
if (binNumber === 1) {
return [0, binWidth];
}
var index = Math.floor(value / binWidth);
return [binWidth * index, binWidth * (index + 1)];
}
// 默认 sturges 转换
function sturges(values) {
return Math.ceil(Math.log(values.length) / Math.LN2) + 1;
}
/**
* 对数据进行百分比化
* @param data
* @param binField
* @param binWidth
* @param binNumber
* @param stackField
*/
function binHistogram(data, binField, binWidth, binNumber, stackField) {
var originData_copy = (0, util_1.clone)(data);
// 根据 binField 对源数据进行排序
(0, util_1.sortBy)(originData_copy, binField);
// 获取源数据 binField 的 range
var values = (0, util_1.valuesOfKey)(originData_copy, binField);
var range = (0, util_1.getRange)(values);
var rangeWidth = range.max - range.min;
// 计算分箱，直方图分箱的计算基于 binWidth，如配置了 binNumber 则将其转为 binWidth 进行计算
var _binWidth = binWidth;
if (!binWidth && binNumber) {
_binWidth = binNumber > 1 ? rangeWidth / (binNumber - 1) : range.max;
}
// 当 binWidth 和 binNumber 都没有指定的情况，采用 Sturges formula 自动生成 binWidth
if (!binWidth && !binNumber) {
var _defaultBinNumber = sturges(values);
_binWidth = rangeWidth / _defaultBinNumber;
}
// 构建 key - StatisticData 结构
var bins = {};
var groups = (0, util_1.groupBy)(originData_copy, stackField);
// 判断分组是否为空，如果为空，说明没有 stackField 字段
if ((0, util_1.isEmpty)(groups)) {
(0, util_1.each)(originData_copy, function (data) {
var value = data[binField];
var bin = getBinKey(value, _binWidth, binNumber);
var binKey = "".concat(bin[0], "-").concat(bin[1]);
if (!(0, util_1.hasKey)(bins, binKey)) {
bins[binKey] = { range: bin, count: 0 };
}
bins[binKey].count += 1;
});
}
else {
Object.keys(groups).forEach(function (groupKey) {
(0, util_1.each)(groups[groupKey], function (data) {
var value = data[binField];
var bin = getBinKey(value, _binWidth, binNumber);
var binKey = "".concat(bin[0], "-").concat(bin[1]);
var groupKeyBinKey = "".concat(binKey, "-").concat(groupKey);
if (!(0, util_1.hasKey)(bins, groupKeyBinKey)) {
bins[groupKeyBinKey] = { range: bin, count: 0 };
bins[groupKeyBinKey][stackField] = groupKey;
}
bins[groupKeyBinKey].count += 1;
});
});
}
// 将分箱数据转换为 plotData 才是图表所需要的
var plotData = [];
(0, util_1.each)(bins, function (bin) {
plotData.push(bin);
});
return plotData;
}
exports.binHistogram = binHistogram;
//# sourceMappingURL=histogram.js.map