Math.Statistics.Dirichlet.Matrix

Implement matrices using plain Vectors with data stored in row-major order (i.e. the first elements correspond to the first row).

Basic

data Matrix

A matrix.

Constructors

M

Fields

mRows :: !Int

mCols :: !Int

mData :: !(Vector Double)

Instances

Eq Matrix

Ord Matrix

Show Matrix

size :: Matrix -> (Int, Int)

Size of the matrix.

(!) :: Matrix -> (Int, Int) -> Double

Element at position.

Constructing

replicate :: (Int, Int) -> Double -> Matrix

A matrix where all elements are of the same value.

replicateRows :: Int -> Vector Double -> Matrix

A matrix where all rows are of the same value.

fromVector :: (Vector v (w Double), Vector w Double) => v (w Double) -> Matrix

Creates a matrix from a vector of vectors. It *is not* verified that the vectors have the right length.

fromVectorT :: (Vector v (w Double), Vector w Double) => v (w Double) -> Matrix

Creates a matrix from a vector of vectors. The vectors are transposed, so fromVectorT is the same as transpose . fromVector. It *is* verified that the vectors have the right length.

Rows

rows :: Matrix -> Vector (Vector Double)

O(rows) Rows of the matrix. Each element takes O(1) time and storage.

(!!!) :: Matrix -> Int -> Vector Double

O(1) m !!! i is the i-th row of the matrix.

Columns

cols :: Matrix -> Vector (Vector Double)

O(rows*cols) Columns of the matrix. Each element takes O(rows) time and storage.

col :: Matrix -> Int -> Vector Double

O(rows) m col i is the i-th column of the matrix.

Maps and zips

umap :: (Vector Double -> Vector Double) -> Matrix -> Matrix

map :: (Double -> Double) -> Matrix -> Matrix

imap :: ((Int, Int) -> Double -> Double) -> Matrix -> Matrix

rowmap :: (Vector Double -> Double) -> Matrix -> Vector Double

irowmap :: (Int -> Vector Double -> Double) -> Matrix -> Vector Double

uzipWith :: (Vector Double -> Vector Double -> Vector Double) -> Matrix -> Matrix -> Matrix

zipWith :: (Double -> Double -> Double) -> Matrix -> Matrix -> Matrix

izipWith :: ((Int, Int) -> Double -> Double -> Double) -> Matrix -> Matrix -> Matrix

rzipWith :: (Int -> Vector Double -> Vector Double -> Vector Double) -> Matrix -> Matrix -> Matrix

rzipWith f m n is a matrix with the same number of rows as m. The i-th row is obtained by applying f to the i-th rows of m and n.

Other

transpose :: Matrix -> Matrix