jyapayne/Nim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Generalize variance to other types

Browse source
This commit is contained in:
Josep Sanjuas 2015-04-18 13:28:48 +02:00
commit e76f54e059
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
lib/pure/math.nim
View file

@ -122,13 +122,13 @@ proc mean*[T](x: openArray[T]): float {.noSideEffect.} =
## ``toFloat(x: T): float`` must be defined.
result = toFloat(sum(x)) / toFloat(len(x))
proc variance*(x: openArray[float]): float {.noSideEffect.} =
proc variance*[T](x: openArray[T]): float {.noSideEffect.} =
## computes the variance of the elements in `x`.
## If `x` is empty, NaN is returned.
result = 0.0
var m = mean(x)
for i in 0 .. high(x):
var diff = x[i] - m
for i in items(x):
var diff = toFloat(i) - m
result = result + diff*diff
result = result / toFloat(len(x))