luciano/xforms
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

x/avg and rfs/avg return nil when no input

Browse source
This commit is contained in:
Christophe Grand 2018-06-13 18:08:34 +02:00
parent c65e766395
commit bae87d1b50
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/net/cgrand/xforms/rfs.cljc
View file

@ -55,11 +55,13 @@
(defn avg (defn avg
"Reducing fn to compute the arithmetic mean." "Reducing fn to compute the arithmetic mean."
([] (transient [0 0])) ([] nil)
([[n sum]] (/ sum n)) ([^doubles acc] (when acc (/ (aget acc 1) (aget acc 0))))
([acc x] (avg acc x 1)) ([acc x] (avg acc x 1))
([[n sum :as acc] x w] ; weighted mean ([^doubles acc x w] ; weighted mean
(-> acc (assoc! 0 (+ n w)) (assoc! 1 (+ sum (* w x)))))) (doto (or acc #?(:clj (double-array 3) :cljs #js [0.0 0.0]))
(aset 0 (+ (aget acc 0) w))
(aset 1 (+ (aget acc 1) (* w x))))))
(defn sd (defn sd
"Reducing fn to compute the standard deviation. Returns 0 if no or only one item." "Reducing fn to compute the standard deviation. Returns 0 if no or only one item."