mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-14 11:23:02 +02:00
acf2f13dcb
* sync with MontFerret/ferret * fix --param handling When params is converted to map it uses strings.Split, which slices a string into all substrings separated by :. * remove impossible conditions nil != nil * delete ineffectual assignments * replace '+= 1' with '++' * remove useless comparison with nil * merge variable declarations * remove bool comparison * fix imports * fix imports * delete unused file * use copy instead of loop * delete unused DummyInterface * remove unnecassary break statements * tidy modules
50 lines
959 B
Go
50 lines
959 B
Go
package math
|
|
|
|
import (
|
|
"context"
|
|
"math"
|
|
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values/types"
|
|
)
|
|
|
|
// Median returns the median of the values in array.
|
|
// @param array (Array) - Array of numbers.
|
|
// @returns (Float) - The median of the values in array.
|
|
func Median(_ context.Context, args ...core.Value) (core.Value, error) {
|
|
err := core.ValidateArgs(args, 1, 1)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
err = core.ValidateType(args[0], types.Array)
|
|
|
|
if err != nil {
|
|
return values.None, err
|
|
}
|
|
|
|
arr := args[0].(*values.Array)
|
|
sorted := arr.Sort()
|
|
|
|
l := sorted.Length()
|
|
|
|
var median core.Value
|
|
|
|
switch {
|
|
case l == 0:
|
|
return values.NewFloat(math.NaN()), nil
|
|
case l%2 == 0:
|
|
median, err = mean(sorted.Slice(l/2-1, l/2+1))
|
|
|
|
if err != nil {
|
|
return values.None, nil
|
|
}
|
|
default:
|
|
median = sorted.Get(l / 2)
|
|
}
|
|
|
|
return median, nil
|
|
}
|