U 3dK@sddlmZddlZddlmZmZ m Z ddl m Z ddZdd Zd d Zd d Zd4ddZddddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd5d*d+Zd6d,d-Z d.d/Z!d0d1Z"d2d3Z#dS)7N)csr_mean_variance_axis0csc_mean_variance_axis0incr_mean_variance_axis0)_check_sample_weightcCs,t|r|jnt|}d|}t|dS)z2Raises a TypeError if X is not a CSR or CSC matrixz,Expected a CSR or CSC sparse matrix, got %s.N)spissparseformattype TypeError)XZ input_typeerrr=/tmp/pip-unpacked-wheel-zrfo1fqw/sklearn/utils/sparsefuncs.py_raise_typeerrorsrcCs|dkrtd|dS)N)rrz8Unknown axis value: %d. Use 0 for rows, or 1 for columns) ValueErroraxisrrr_raise_error_wrong_axissrcCs6|jd|jdkst|j|j|jdd9_dS)a Inplace column scaling of a CSR matrix. Scale each feature of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to normalize using the variance of the features. It should be of CSR format. scale : ndarray of shape (n_features,), dtype={np.float32, np.float64} Array of precomputed feature-wise values to use for scaling. rrZclip)modeN)shapeAssertionErrordataZtakeindicesr Zscalerrrinplace_csr_column_scalesrcCs:|jd|jdkst|jt|t|j9_dS)aInplace row scaling of a CSR matrix. Scale each sample of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to be scaled. It should be of CSR format. scale : ndarray of float of shape (n_samples,) Array of precomputed sample-wise values to use for scaling. rN)rrrnprepeatdiffindptrrrrrinplace_csr_row_scale2sr!FcCs|t|t|tjr<|dkr*t|||dSt|j||dSnt|tjrt|j|n t|tjr2t||nt|dS)aInplace column scaling of a CSC/CSR matrix. Scale each feature of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to normalize using the variance of the features. It should be of CSC or CSR format. scale : ndarray of shape (n_features,), dtype={np.float32, np.float64} Array of precomputed feature-wise values to use for scaling. N)r$rr)r!r(r%rrrrrrinplace_column_scales    r3cCs>t|tjrt|j|n t|tjr2t||nt|dS)aInplace row scaling of a CSR or CSC matrix. Scale each row of the data matrix by multiplying with specific scale provided by the caller assuming a (n_samples, n_features) shape. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix to be scaled. It should be of CSR or CSC format. scale : ndarray of shape (n_features,), dtype={np.float32, np.float64} Array of precomputed sample-wise values to use for scaling. N)r$rr)rr(r%r!rrrrrinplace_row_scales    r4cCsv||fD]}t|tjrtdq|dkr8||jd7}|dkrN||jd7}|j|k}||j|j|k<||j|<dS)aKSwap two rows of a CSC matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two rows are to be swapped. It should be of CSC format. m : int Index of the row of X to be swapped. n : int Index of the row of X to be swapped. m and n should be valid integersrN)r$rndarrayr rr)r mntZm_maskrrrinplace_swap_row_csc s    r:c Csx||fD]}t|tjrtdq|dkr8||jd7}|dkrN||jd7}||kr`||}}|j}||}||d}||}||d}||} ||} | | kr|j|d|| | 7<|| |j|d<|| |j|<t|jd||j|||j|||j|||j|dg|_t|jd||j|||j|||j|||j|dg|_dS)aKSwap two rows of a CSR matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two rows are to be swapped. It should be of CSR format. m : int Index of the row of X to be swapped. n : int Index of the row of X to be swapped. r5rrrN) r$rr6r rr Z concatenaterr) r r7r8r9r Zm_startZm_stopZn_startZn_stopZnz_mZnz_nrrrinplace_swap_row_csr(sH                 r;cCs@t|tjrt|||n"t|tjr4t|||nt|dS)a[ Swap two rows of a CSC/CSR matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two rows are to be swapped. It should be of CSR or CSC format. m : int Index of the row of X to be swapped. n : int Index of the row of X to be swapped. N)r$rr)r:r%r;rr r7r8rrrinplace_swap_rowgs   r=cCsl|dkr||jd7}|dkr,||jd7}t|tjrFt|||n"t|tjr`t|||nt|dS)ag Swap two columns of a CSC/CSR matrix in-place. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Matrix whose two columns are to be swapped. It should be of CSR or CSC format. m : int Index of the column of X to be swapped. n : int Index of the column of X to be swapped. rrN)rr$rr)r;r%r:rr<rrrinplace_swap_columns  r>cCsLtt|j}t||j|j|jf|jd}||j|j|}||fS)N)r) rZ flatnonzerorr r rrrZreduceat)r Zufunc major_indexvaluerrr _minor_reducesrAc Cs|j|}|dkrtd|jd|}|dkr8|n|}|t||\}}t|j||k}|||d||<|dk} t | |}t | |}|dkrt j |t t ||ff|jd|fd} n(t j ||t t |ff|j|dfd} | jS)Nr&zero-size array to reduction operationr)r+r)rrZtocscZtocsrZsum_duplicatesrArrr compressrZ coo_matrixzeroslenr+Aravel) r r min_or_maxNMmatr?r@not_fullmaskresrrr_min_or_max_axiss0   rOcCs|dkrdd|jkrtd|jd}|jdkr4|S||j}|jt |jkr`|||}|S|dkrt|d7}|dks|dkrt |||StddS)NrrBrrz.invalid axis, use 0 for rows, or 1 for columns) rrr+r nnzreducerrGrproductrO)r rrHZzeror7rrr_sparse_min_or_maxs     rScCst||tjt||tjfSN)rSrZminimummaximumr rrrr_sparse_min_maxs  rWcCst||tjt||tjfSrT)rSrZfminZfmaxrVrrr_sparse_nan_min_maxsrXcCs<t|tjtjfr0|r"t||dSt||dSnt|dS)aCompute minimium and maximum along an axis on a CSR or CSC matrix. Optionally ignore NaN values. Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Input data. It should be of CSR or CSC format. axis : {0, 1} Axis along which the axis should be computed. ignore_nan : bool, default=False Ignore or passing through NaN values. .. versionadded:: 0.20 Returns ------- mins : ndarray of shape (n_features,), dtype={np.float32, np.float64} Feature-wise minima. maxs : ndarray of shape (n_features,), dtype={np.float32, np.float64} Feature-wise maxima. rN)r$rr%r)rXrWr)r rZ ignore_nanrrr min_max_axiss  rYcCs|dkrd}n(|dkrd}n|jdkr6td|j|dkrb|dkrL|jStt|j|Sn|dkrt|j}|dkr|dS||S|dkr|dkrtj|j |j dd St |t|j}tj|j |j d|d Snt d |dS) aA variant of X.getnnz() with extension to weighting on axis 0. Useful in efficiently calculating multilabel metrics. Parameters ---------- X : sparse matrix of shape (n_samples, n_labels) Input data. It should be of CSR format. axis : {0, 1}, default=None The axis on which the data is aggregated. sample_weight : array-like of shape (n_samples,), default=None Weight for each row of X. Returns ------- nnz : int, float, ndarray of shape (n_samples,) or ndarray of shape (n_features,) Number of non-zero values in the array along a given axis. Otherwise, the total number of non-zero values in the array is returned. rrZcsrz#Expected CSR sparse format, got {0}NZintp) minlength)r\r"zUnsupported axis: {0}) r r rPrdotrr ZastypeZbincountrrrr)r rZ sample_weightoutr"rrr count_nonzeros*   r_cCspt||}|stjSt|dk}t|d\}}||rLt||||St|d|||t||||dS)zCompute the median of data with n_zeros additional zeros. This function is used to support sparse matrices; it modifies data in-place. rrrg@)rErnanr_divmodsort_get_elem_at_rank)rn_zerosZn_elems n_negativeZmiddleZis_oddrrr _get_median7s  rfcCs,||kr||S|||kr dS|||S)z@Find the value in data augmented with n_zeros for the given rankrr)ZrankrrerdrrrrcMs  rcc Cst|tjstd|j|j}|j\}}t|}t t |dd|ddD]8\}\}}t |j ||}||j } t|| ||<qR|S)aCFind the median across axis 0 of a CSC matrix. It is equivalent to doing np.median(X, axis=0). Parameters ---------- X : sparse matrix of shape (n_samples, n_features) Input data. It should be of CSC format. Returns ------- median : ndarray of shape (n_features,) Median. z%Expected matrix of CSC format, got %sNrZr)r$rr)r r r rrrD enumeratezipcopyrr/rf) r r Z n_samplesZ n_featuresZmedianZf_indstartendrZnzrrrcsc_median_axis_0Vs   * rl)NF)F)NN)$Z scipy.sparsesparserZnumpyrZsparsefuncs_fastrr&rr'rr1Zutils.validationrrrrr!r*r2r3r4r:r;r=r>rArOrSrWrXrYr_rfrcrlrrrrs2   <^?  $ 6