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

math.matrics: Add some combinators and indexing words to math.matrices.

db4
Doug Coleman 2012-10-02 17:53:24 -07:00
parent 3582a6c624
commit b23e52f532
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
basis/math/matrices/matrices.factor
View File

@ -168,3 +168,31 @@ IN: math.matrices
: outer ( u v -- m )
[ n*v ] curry map ;
: row ( n m -- col )
nth ; inline
: rows ( seq m -- cols )
'[ _ row ] map ; inline
: col ( n m -- col )
swap '[ _ swap nth ] map ; inline
: cols ( seq m -- cols )
'[ _ col ] map ; inline
: matrix-map ( m quot -- )
'[ _ map ] map ; inline
: column-map ( m quot -- seq )
[ [ first length iota ] keep ] dip '[ _ col @ ] map ; inline
: cartesian-indices ( n -- matrix )
iota dup cartesian-product ; inline
: cartesian-matrix-map ( m quot -- m' )
[ [ first length cartesian-indices ] keep ] dip
'[ _ @ ] matrix-map ; inline
: cartesian-matrix-column-map ( m quot -- m' )
[ cols first2 ] prepose cartesian-matrix-map ; inline