diff options
author | Pauli Virtanen <pav@iki.fi> | 2008-10-28 00:13:44 +0000 |
---|---|---|
committer | Pauli Virtanen <pav@iki.fi> | 2008-10-28 00:13:44 +0000 |
commit | 18594cd9653a865fddfa4cd81f82ab54430be1c9 (patch) | |
tree | 04db708f8a8a3575d129390342ff789ef6f1e170 /numpy/matlib.py | |
parent | 7a70f54f515bb8c586c3967d62731a49217eef95 (diff) | |
download | numpy-18594cd9653a865fddfa4cd81f82ab54430be1c9.tar.gz |
Import documentation from doc wiki (part 2, work-in-progress docstrings, but they are still an improvement)
Diffstat (limited to 'numpy/matlib.py')
-rw-r--r-- | numpy/matlib.py | 141 |
1 files changed, 138 insertions, 3 deletions
diff --git a/numpy/matlib.py b/numpy/matlib.py index 4ffc0ba13..7026f5c1a 100644 --- a/numpy/matlib.py +++ b/numpy/matlib.py @@ -14,21 +14,123 @@ def empty(shape, dtype=None, order='C'): return ndarray.__new__(matrix, shape, dtype, order=order) def ones(shape, dtype=None, order='C'): - """return a matrix initialized to all ones + """ + Matrix of ones. + + Return a matrix of given shape and type, filled with ones. + + Parameters + ---------- + shape : {sequence of ints, int} + Shape of the matrix + dtype : data-type, optional + The desired data-type for the matrix, default is np.float64. + order : {'C', 'F'}, optional + Whether to store matrix in C- or Fortran-contiguous order, + default is 'C'. + + Returns + ------- + out : matrix + Matrix of ones of given shape, dtype, and order. + + See Also + -------- + ones : Array of ones. + matlib.zeros : Zero matrix. + + Notes + ----- + If `shape` has length one i.e. ``(N,)``, or is a scalar ``N``, + `out` becomes a single row matrix of shape ``(1,N)``. + + Examples + -------- + >>> np.matlib.ones((2,3)) + matrix([[ 1., 1., 1.], + [ 1., 1., 1.]]) + + >>> np.matlib.ones(2) + matrix([[ 1., 1.]] + """ a = ndarray.__new__(matrix, shape, dtype, order=order) a.fill(1) return a def zeros(shape, dtype=None, order='C'): - """return a matrix initialized to all zeros + """ + Zero matrix. + + Return a matrix of given shape and type, filled with zeros + + Parameters + ---------- + shape : {sequence of ints, int} + Shape of the matrix + dtype : data-type, optional + The desired data-type for the matrix, default is np.float64. + order : {'C', 'F'}, optional + Whether to store the result in C- or Fortran-contiguous order, + default is 'C'. + + Returns + ------- + out : matrix + Zero matrix of given shape, dtype, and order. + + See Also + -------- + zeros : Zero array. + matlib.ones : Matrix of ones. + + Notes + ----- + If `shape` has length one i.e. ``(N,)``, or is a scalar ``N``, + `out` becomes a single row matrix of shape ``(1,N)``. + + Examples + -------- + >>> np.matlib.zeros((2,3)) + matrix([[ 0., 0., 0.], + [ 0., 0., 0.]]) + + >>> np.matlib.zeros(2) + matrix([[ 0., 0.]] + """ a = ndarray.__new__(matrix, shape, dtype, order=order) a.fill(0) return a def identity(n,dtype=None): - """identity(n) returns the identity matrix of shape n x n. + """ + Returns the square identity matrix of given size. + + Parameters + ---------- + n : int + Size of identity matrix + + dtype : data-type, optional + Data-type of the output. Defaults to ``float``. + + Returns + ------- + out : matrix + `n` x `n` matrix with its main diagonal set to one, + and all other elements zero. + + See Also + -------- + identity : Equivalent array function. + matlib.eye : More general matrix identity function. + + Notes + ----- + For more detailed documentation, see the docstring of the equivalent + array function ``np.identity`` + """ a = array([1]+n*[0],dtype=dtype) b = empty((n,n),dtype=dtype) @@ -36,6 +138,39 @@ def identity(n,dtype=None): return b def eye(n,M=None, k=0, dtype=float): + """ + Return a matrix with ones on the diagonal and zeros elsewhere. + + Parameters + ---------- + n : int + Number of rows in the output. + M : int, optional + Number of columns in the output, defaults to n. + k : int, optional + Index of the diagonal: 0 refers to the main diagonal, + a positive value refers to an upper diagonal, + and a negative value to a lower diagonal. + dtype : dtype, optional + Data-type of the returned matrix. + + Returns + ------- + I : matrix + A `n` x `M` matrix where all elements are equal to zero, + except for the k-th diagonal, whose values are equal to one. + + See Also + -------- + eye : Equivalent array function + matlib.identity : Square identity matrix + + Notes + ----- + For more detailed docuemtation, see the docstring of the equivalent + array function ``np.eye``. + + """ return asmatrix(np.eye(n,M,k,dtype)) def rand(*args): |