a CCCf8@sdZddlZddlmZmZmZmZddlZddlZ ddl m Z ddl m ZddlmZmZgdZe je je je je je je jeee je je je je je je j e j!gZ"iZ#ddZ$d d Z%d d Z&d dZ'ddZ(d>ddZ)d?e j*dddZ+d@ddZ,e j-e j-dddZ.e/dddZ0e/ddd Z1dAdd!e/dd"d#Z2e/dd$d%Z3e/dd&d'Z4e/dd(d)Z5ddd*d+Z6dBdd!e7e8d,fdd-d.Z9d/d0Z:e/dd1d2Z;dCeeed3eed4fd5d6d7Ze j*dd T where T is a supported dtype Examples -------- >>> from scipy.sparse._sputils import upcast >>> upcast('int32') >>> upcast('bool') >>> upcast('int32','float32') >>> upcast('bool',complex,float) Nz#no supported conversion for types: ) _upcast_memogethashnpZ result_typesupported_dtypescan_cast TypeError)argstr rQ/var/www/html/django/DPS/env/lib/python3.9/site-packages/scipy/sparse/_sputils.pyr s    r cGs2t|}|dur|Stttj|}|t|<|S)z9Same as `upcast` but taking dtype.char as input (faster).N)rrr maprdtype)rrrrr upcast_char:s  r cCstjdg|d|jS)z`Determine data type for binary operation between an array of type `dtype` and a scalar. rr)rarrayr)rZscalarrrr upcast_scalarDsr#cCsx|jjttjjkrt|jdkr,|tjS|}|}|ttjjks`|ttjjkrht d|tjS|S)z Down-cast index array to np.intp dtype if it is of a larger dtype. Raise an error if the array contains a value that is too large for intp. rzzCannot deal with arrays with indices larger than the machine maximum address size (e.g. 64-bit indices on 32-bit machine).) ritemsizerZintpsizeZastypemaxminiinfo ValueError)arrmaxvalminvalrrrdowncast_intp_indexKs  $ r-cCs$|j}|jr|Stj||ddS)a Ensure that the data type of the NumPy array `A` has native byte order. `A` must be a NumPy array. If the data type of `A` does not have native byte order, a copy of `A` with a native byte order is returned. Otherwise `A` is returned. nativer!)rZisnativerasarrayZ newbyteorder)Adtrrr to_native_sr2c Csx|durXz |j}WqttyT}z*|dur6t|}n td|WYd}~qtd}~00nt|}|tjkrttd|S)aQFunction used to simplify argument processing. If 'dtype' is not specified (is None), returns a.dtype; otherwise returns a np.dtype object created from the specified dtype argument. If 'dtype' and 'a' are both None, construct a data type out of the 'default' parameter. Furthermore, 'dtype' must be in 'allowed' set. Nzcould not interpret data typez0object dtype is not supported by sparse matrices)rAttributeErrorrrZobject_r))radefaultZnewdtypeerrrr os  "  r F)returncCstj|||d}t|j|S)z This is a wrapper of `np.array(obj, dtype=dtype, copy=copy)` that will generate a warning if the result is an object array. )rcopy)rr"r r)objrr8datarrrr s r rcCstttjj}tttjj}tjdkrt|ttfr(t|dkprFrrrrrs rcCs<t|ttfr&t|dkr&t|dp:t|tjo:|jdkS)NrrO)r=rSrTrQrrr>rFrUrrrrs  rcCs t|tjSN)r=rr>rGrrrr srcCsj|dur dSt|}|tkr$tdtt|tjsHtd|jd|kr\dksfntddS)NzlTuples are not accepted for the 'axis' parameter. Please pass in one of the following: {-2, -1, 0, 1, None}.zaxis must be an integer, not rPzaxis out of range) typerTrrr?rr@__name__r))ZaxisZ axis_typerrr validateaxissrZ.c Cst|dkrtdt|dkrhzt|d}Wn"tyRt|df}Yqz0tdd|D}ntdd|D}|dur|rt|dvrtd nt|d krtd td d|Drtd nt|}ddt |D}|st|}||krtd ||nt|dkr|d}t|d|||dd} t || \} } | dkrtdd|D} td || |d|| f||dd}ntdt|d kr|rt|dkstd|S)auImitate numpy.matrix handling of shape arguments Parameters ---------- args : array_like Data structures providing information about the shape of the sparse array. current_shape : tuple, optional The current shape of the sparse array or matrix. If None (default), the current shape will be inferred from args. allow_1d : bool, optional If True, then 1-D or 2-D arrays are accepted. If False (default), then only 2-D arrays are accepted and an error is raised otherwise. Returns ------- new_shape: tuple The new shape after validation. rz8function missing 1 required positional argument: 'shape'rPcss|]}t|VqdSrVrIrJ.0argrrr Ezcheck_shape..css|]}t|VqdSrVr[r\rrrr_Gr`N)rPrOz2shape must be a 1- or 2-tuple of positive integersrOz,shape must be a 2-tuple of positive integerscss|]}|dkVqdS)rNr)r]rRrrrr_Pr`z#'shape' elements cannot be negativecSsg|]\}}|dkr|qS)rr)r]irHrrr Wr`zcheck_shape..z-cannot reshape array of size {} into shape {}css|]}|dkrdn|VqdS)rZnewshapeNr)r]rHrrrr_br`z&can only specify one unknown dimensionz$matrix shape must be two-dimensional) rQriterrIrJrTr)anyr enumerateformatdivmod) rZ current_shaperNZ shape_iterZ new_shapeZ current_sizeZnegative_indexesnew_sizeskipZ specified unspecified remainderZ err_shaperrr check_shape(sN          $"rlcCs<|dd}|dd}|r4tdd|||fS)a?Unpack keyword arguments for reshape function. This is useful because keyword arguments after star arguments are not allowed in Python 2, but star keyword arguments are. This function unpacks 'order' and 'copy' from the star keyword arguments (with defaults) and throws an error for any remaining. orderCr8Fz/reshape() got unexpected keywords arguments: {}z, )poprrfjoinkeys)kwargsrmr8rrrcheck_reshape_kwargsos   rscCs&ttjddd}|duo$t||S)zV Check whether object is pydata/sparse matrix, avoiding importing the module. sparseZ SparseArrayN)getattrsysmodulesrr=)mbase_clsrrris_pydata_spmatrixsrzZcscZcsrz sp.spmatrix)r^ target_formatr7cCs:t|r6|}|dur$||}n|jdvr6|}|S)z_ Convert a pydata/sparse array to scipy sparse matrix, pass through anything else. Nr{)rzZto_scipy_sparseZasformatrfZtocsc)r^r|rrrconvert_pydata_sparse_to_scipys  r}cOstj|i|tjSrV)rr"viewmatrix)rrrrrrrsrcCs8t|tjr"|dus|j|kr"|Stj||dtjS)Nr!)r=rrrr/r~)r:rrrrasmatrixsrcCsxt|tjjr|St|tjr:tj||j |j dSt|tj rltj |j |j d}tj |j||S|S)zAccess nonzero values, possibly after summing duplicates. Parameters ---------- s : sparse array Input sparse array. Returns ------- data: ndarray Nonzero values of the array, with shape (s.nnz,) )rcountr!)r=sp_dataZ _data_matrixZ _deduped_dataZ dok_arrayrZfromitervaluesrZnnzZ lil_arrayemptyZ _csparsetoolsZlil_flatten_to_arrayr:Ztocoo)sr:rrr_todatas  r)NN)NF)rNF)F)N)N)N)@__doc__rvtypingrrrrrInumpyrmathrZ scipy.sparsertrZscipy._lib._utilrr__all__Zbool_byteZubyteshortZushortr<ZuintcZlonglongZ ulonglongZfloat32Zfloat64Z longdoubleZ complex64Z complex128Z clongdoublerrr r r#r-r2r r>r rArrrKr r rrrrrZrTrLrlrsrzr}rrrrrrrsR    "   7 G