mirror of
https://github.com/MontFerret/ferret.git
synced 2025-03-19 21:28:32 +02:00
24d8eedd4c
* Added release notes * #509 fixedOCOD typo * Updated values * Updated comments * Changed stdlib docs format * Changed format of array in docs * Use 'any' instead of 'value' in docs * New format for optional params * Updated docs for testing package * Added namespace information
50 lines
968 B
Go
50 lines
968 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 {Int[] | Float[]} array - Array of numbers.
|
|
// @return {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
|
|
}
|