correctly handle categoricals that are not strings (#1346)

This commit is contained in:
Bruce Martin
2020-04-03 15:50:02 -07:00
committed by GitHub
parent 4117050081
commit de8033e551
2 changed files with 10 additions and 6 deletions
+8 -4
View File
@@ -12,7 +12,7 @@ import {
summarizeCategorical as _summarizeCategorical
} from "./summarize";
import {
histogramCategorical,
histogramCategorical as _histogramCategorical,
hashCategorical,
histogramContinuous,
hashContinuous
@@ -260,16 +260,20 @@ class Dataframe {
/*
Create histogram bins for this column. Memoized.
*/
const _memoHistoCat = memoize(_histogramCategorical, hashCategorical);
const histogramCategorical = by => _memoHistoCat(get, by);
let histogram = null;
if (isTypedArray(column)) {
const mFn = memoize(histogramContinuous, hashContinuous);
get.histogram = (bins, domain, by) => mFn(get, bins, domain, by);
histogram = (bins, domain, by) => mFn(get, bins, domain, by);
} else {
const mFn = memoize(histogramCategorical, hashCategorical);
get.histogram = by => mFn(get, by);
histogram = histogramCategorical;
}
get.summarize = summarize;
get.summarizeCategorical = summarizeCategorical;
get.histogram = histogram;
get.histogramCategorical = histogramCategorical;
get.asArray = asArray;
get.has = has;
get.ihas = ihas;