:j ^dZdZgdZddlZddlmZddlmZmZddl m Z m Z d Z d d Z d d Zy) z.Functions to extract parts of sparse matrices zrestructuredtext en)findtriltriuN)warn) coo_array coo_matrix)sparrayspmatrixct|d}|j|jdk7}|j||j||j|fS)aUReturn the indices and values of the nonzero elements of a matrix. Parameters ---------- A : dense or sparse array or matrix Matrix whose nonzero elements are desired. Returns ------- (I,J,V) : tuple of arrays I,J, and V contain the row indices, column indices, and values of the nonzero entries. Examples -------- >>> from scipy.sparse import csr_array, find >>> A = csr_array([[7.0, 8.0, 0],[0, 0, 9.0]]) >>> find(A) (array([0, 0, 1], dtype=int32), array([0, 1, 2], dtype=int32), array([ 7., 8., 9.])) Tcopyr)rsum_duplicatesdatarowcol)Anz_masks j/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/sparse/_extract.pyrrsN4 !$AffkG 55>155>166'? ::ct|trt}nUt|trt}n>d}t j jtf}t|t|t}||d}|j|z|jk\}|j|}|j|}|j|} || ||ff|j|j} | j!|S)aReturn the lower triangular portion of a sparse array or matrix. Returns the elements on or below the k-th diagonal of A. - k = 0 corresponds to the main diagonal - k > 0 is above the main diagonal - k < 0 is below the main diagonal .. warning:: `tril` is switching to the sparse array interface. For the case where no input arrays are sparse, this function is switching to returning a sparse array instead of sparse matrix. Control the sparse return class by making at least one input sparse, e.g., ``tril(coo_matrix(A))``, or ``tril(coo_array(A))``. That removes any deprecation warnings as well. For more general information about sparrays, see :ref:`Migration from spmatrix to sparray `. Handling of this no sparse input case will change no earlier than v1.20. Parameters ---------- A : dense or sparse array or matrix Matrix whose lower trianglar portion is desired. k : int : optional The top-most diagonal of the lower triangle. format : str Sparse format of the result, e.g. format="csr", etc. Returns ------- L : sparse matrix Lower triangular portion of A in sparse format. See Also -------- triu : upper triangle in sparse format Examples -------- >>> from scipy.sparse import csr_array, tril >>> A = csr_array([[1, 2, 0, 0, 3], [4, 5, 0, 6, 7], [0, 0, 8, 9, 0]], ... dtype='int32') >>> A.toarray() array([[1, 2, 0, 0, 3], [4, 5, 0, 6, 7], [0, 0, 8, 9, 0]], dtype=int32) >>> tril(A).toarray() array([[1, 0, 0, 0, 0], [4, 5, 0, 0, 0], [0, 0, 8, 0, 0]], dtype=int32) >>> tril(A).nnz 4 >>> tril(A, k=1).toarray() array([[1, 2, 0, 0, 0], [4, 5, 0, 0, 0], [0, 0, 8, 9, 0]], dtype=int32) >>> tril(A, k=-1).toarray() array([[0, 0, 0, 0, 0], [4, 0, 0, 0, 0], [0, 0, 0, 0, 0]], dtype=int32) >>> tril(A, format='csc') ay`tril` is switching to the sparse array interface. For the case where input arrays are numpy arrays, this function is switching to returning a sparse array instead of sparse matrix. Recover the sparse matrix return value by making one input a sparse matrix. For example, tril(coo_matrix(A)). Avoid this message for sparse array output by using tril(coo_array(A)). For more information, see the spmatrix to sparray migration guide https://docs.scipy.org/doc/scipy/reference/sparse.migration_to_sparray.html This function will be changed no earlier than v1.20. categoryskip_file_prefixesFr shapedtype isinstancer rr r ospathdirname__file__rDeprecationWarningrrrrrasformat rkformat coo_sparsemsgprefixesmaskrrrnew_coos rrr0sF!W Ax    GGOOH-/ S-(K  15!A 5519 D %%+C %%+C 66$d}t j jtf}t|t|t}||d}|j|z|jk}|j|}|j|}|j|} || ||ff|j|j} | j!|S)aReturn the upper triangular portion of a sparse array or matrix. Returns the elements on or above the k-th diagonal of A. - k = 0 corresponds to the main diagonal - k > 0 is above the main diagonal - k < 0 is below the main diagonal .. warning:: `triu` is switching to the sparse array interface. For the case where no input arrays are sparse, this function is switching to returning a sparse array instead of sparse matrix. Control the sparse return class by making at least one input sparse, e.g., ``triu(coo_matrix(A))``, or ``triu(coo_array(A))``. That removes any deprecation warnings as well. For more general information about sparrays, see :ref:`Migration from spmatrix to sparray `. Handling of this no sparse input case will change no earlier than v1.20. Parameters ---------- A : dense or sparse array or matrix Matrix whose upper trianglar portion is desired. k : int : optional The bottom-most diagonal of the upper triangle. format : str Sparse format of the result, e.g. format="csr", etc. Returns ------- L : sparse array or matrix Upper triangular portion of A in sparse format. Sparse array if A is a sparse array, otherwise matrix. See Also -------- tril : lower triangle in sparse format Examples -------- >>> from scipy.sparse import csr_array, triu >>> A = csr_array([[1, 2, 0, 0, 3], [4, 5, 0, 6, 7], [0, 0, 8, 9, 0]], ... dtype='int32') >>> A.toarray() array([[1, 2, 0, 0, 3], [4, 5, 0, 6, 7], [0, 0, 8, 9, 0]], dtype=int32) >>> triu(A).toarray() array([[1, 2, 0, 0, 3], [0, 5, 0, 6, 7], [0, 0, 8, 9, 0]], dtype=int32) >>> triu(A).nnz 8 >>> triu(A, k=1).toarray() array([[0, 2, 0, 0, 3], [0, 0, 0, 6, 7], [0, 0, 0, 9, 0]], dtype=int32) >>> triu(A, k=-1).toarray() array([[1, 2, 0, 0, 3], [4, 5, 0, 6, 7], [0, 0, 8, 9, 0]], dtype=int32) >>> triu(A, format='csc') ay`triu` is switching to the sparse array interface. For the case where input arrays are numpy arrays, this function is switching to returning a sparse array instead of sparse matrix. Recover the sparse matrix return value by making one input a sparse matrix. For example, triu(coo_matrix(A)). Avoid this message for sparse array output by using triu(coo_array(A)). For more information, see the spmatrix to sparray migration guide https://docs.scipy.org/doc/scipy/reference/sparse.migration_to_sparray.html This function will be changed no earlier than v1.20. rFr rrr&s rrrsH!W Ax    GGOOH-/ S-(K  15!A 5519 D %%+C %%+C 66$r6s5& " '$;Ba$Hb$r