steve
/
factor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

math.statistics: add rank-values

db4
Doug Coleman 2012-09-04 18:52:22 -07:00
parent 8ed541b26b
commit d2317c71f5
2 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
basis/math/statistics/statistics-tests.factor
View File

@ -157,3 +157,15 @@ IN: math.statistics.tests
] unit-test ] unit-test
{ { 0 1/4 1/2 3/4 1 } } [ 5 iota rescale ] unit-test { { 0 1/4 1/2 3/4 1 } } [ 5 iota rescale ] unit-test
{
{ 2 2 2 1 0 5 6 7 7 7 7 }
} [
{ 30 30 30 20 10 40 50 60 60 60 60 } rank-values
] unit-test
{
{ 1 0 2 3 4 }
}
[ { 3 1 4 15 92 } rank-values ] unit-test

11
basis/math/statistics/statistics.factor
View File

@ -2,7 +2,8 @@
! See http://factorcode.org/license.txt for BSD license. ! See http://factorcode.org/license.txt for BSD license.
USING: assocs combinators generalizations kernel locals math USING: assocs combinators generalizations kernel locals math
math.functions math.order math.vectors sequences math.functions math.order math.vectors sequences
sequences.private sorting fry arrays grouping sets ; sequences.private sorting fry arrays grouping sets
splitting.monotonic ;
IN: math.statistics IN: math.statistics
: power-mean ( seq p -- x ) : power-mean ( seq p -- x )
@ -343,3 +344,11 @@ ALIAS: corr sample-corr
: rescale ( u -- v ) : rescale ( u -- v )
dup minmax over - [ v-n ] [ v/n ] bi* ; dup minmax over - [ v-n ] [ v/n ] bi* ;
: rank-values ( seq -- seq' )
[
[ ] [ length iota ] bi zip sort-keys
[ [ first ] bi@ = ] monotonic-split
[ values ] map [ 0 [ length + ] accumulate nip ] [ ] bi zip
] [ length f <array> ] bi
[ '[ first2 [ _ set-nth ] with each ] each ] keep ;