wg:j dZddlmZmZmZmZmZmZmZm Z ddlm Z ddl Z ddl Z ddl mZmZmZddlmZmZmZe j0j2j4Ze j6j8ZgdZgdZddZd Z Gd d eZ!d Z"e!Z# ddZdddddd ddef dZdZ$dZ% de jLddZ'ddZ(y)a,:mod:`numpy.ma..mrecords` Defines the equivalent of :class:`numpy.recarrays` for masked arrays, where fields can be accessed as attributes. Note that :class:`numpy.ma.MaskedArray` already supports structured datatypes and the masking of individual fields. .. moduleauthor:: Pierre Gerard-Marchant )MAError MaskedArraymaskednomask masked_arraygetdata getmaskarrayfilledN)dtypendarrayarray)recarray fromarrays fromrecords) MaskedRecords mrecarrayrr fromtextfileaddfield)_data_mask _fieldmaskr c6t|}t|Dcgc]}d|z }}||}nat|ttfr|}n0t|t r|j d}ntd|t|}||kr|||dz }g}t|||jD]X\}}} |tvr4| dtvr|j|| df1|j| C|j|| dfZtj|Scc}w)z Checks that field names ``descr`` are not reserved keywords. If this is the case, a default 'f%i' is substituted. If the argument `names` is not None, updates the field names to valid names. f%iN,zillegal input names r)lenrange isinstancetupleliststrsplit NameErrorzipdescrreserved_fieldsappendnpr ) r%namesndescri default_names new_namesnnamesndts V/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numpy/ma/mrecords.py _checknamesr3,sZF(-f 61UQY6M6 }! eeT] +I s # C(I25)<= =Y F? vw/ /I FM5;;?% Aq  t& q!A$i( a MM1ad) $% 88F -7s Dc|jjDcgc]}|df}}tj|j|}t dgt |z|_|Scc}w)Nz|b1r F)r r)r(emptyshaperrflat)selfr/mdescrfdmasks r2_get_fieldmaskr<NsX"&**"2"2 3Qq%j 3F 3 XXdjj /F#f+-.FK M4s A&ceZdZdZdddddddddeddddfdZdZedZed Z d Z d Z d Z d Z dZdZdZddZdZdZdZddZdZdZdZy)ra Attributes ---------- _data : recarray Underlying data, as a record array. _mask : boolean array Mask of the records. A record is masked when all its fields are masked. _fieldmask : boolean recarray Record array of booleans, setting the mask of each individual field of each record. _fill_value : record Filling values for each field. NrFTc 6tj|||||||||| |  }tj|j}| t ust j| s!|stdgt|z|_ |St j| |} | j|jk7rv|j| j}}|dk(r!t j| |j} n8||k(r!t j| |j} nd}t|||fz|s|j!| d|_|S| j|k(r| }n;t j| Dcgc]}t|gt|zc}|}||_ |Scc}w)N) r bufoffsetstridesformatsr)titles byteorderalignedF)copyrz?Mask and data not compatible: data size is %i, mask size is %i.Tr5)r__new__mamake_mask_descrr rr(sizerrrr r7resizereshaper __setmask__ _sharedmask)clsr7r r?r@rArBr)rCrDrEmask hard_mask fill_value keep_maskrFoptionsr9mdtypendnmmsgrms r2rGzMaskedRecords.__new__gskU%S(/'-(/2 ##DJJ/ 6>"E7S[#89 . +88Dt,DzzTZZ' IItyyR799T4::6D2X::dDJJ7D-C!#R.11  &#'  ::' EHH%M1eQC#f+,=&>%M+13E"  &Ns%Fc t|dd}|t|dt}tj|d}|tur"t j |j |}nYt j|}t|Dcgc]}t|gt|zc}|jt}|j}|j||j||dtk(r t|d<ycc}w)Nrr r5r _baseclass)getattrrr __getattribute__rHmake_mask_noner7rInarrayrrviewr__dict__update _update_from)r9objrobjmask_dtyper:rY_dicts r2__array_finalize__z MaskedRecords.__array_finalize__sWd+ =c7F3G--dG? @ @--dJ?  T:l#;<ennc" 99?? &%'89 9 w-   d   %F DF9B#HIMMOI II((;'C$CNCLCI$..=K&+&1$&7CO  "+&*CO + ((*C sH $$AA4 A//A4-C33 C?>C? FF21F2c&|dvr|j|ytj|d}||v} tj|||}t j|dj xsi}||vr|S|r tj || ||dd}|tur|d} | |d|} n|} d } nt|} t|} t j|d j| g|} |d j|| | S#t$r|cYSwxYw#t$rJt j|dj xsi}t j|dxsi}||vs||vsYwxYw#ttf$r} td|| d} ~ wwxYw) z< Sets the attribute attr to the value val. )rP fieldmaskNrbr _optinforrrsruTrr)rMrvr^ __setattr__r rx __delattr__ Exceptionryrzrwrr r setfield __setitem__)r9rvalrnewattrretroptinforrrudvalmvalres r2rzMaskedRecords.__setattr__s ( (   S ! ,,T:> j( $$T45C 00w?FFL"I9$ &&tT2 @D/"1%C &=$]3K&!-06D#;D$D>g&&tW5>>tJcJ7''d3 +!J# 00w?FFL"I..tZ@FBGI%  ,8$ @ 079>? @ @s<D6D E- DDAE*)E*-F<F  Fc|j}tj|d}tj||d}t |t rZ||jt }|||_d|_|d}| |||_ |js|jrtS|Stj||jt}tj||jt|_|S)z Returns all the fields sharing the same fieldname base. The fieldname base is either `_data` or `_mask`. rr\Tru)rbr r^rarr!rrrNrurorr(asarrayrr)r9indxrrrrefvals r2 __getitem__zMaskedRecords.__getitem__%s]] ((w7 T:l#;< dC +"";/Cd CI"COm,D"&t*88  Jjjt%**95JJuT{+00:  rlctj|||t|tr#t j ||j |<yy)z2 Sets the given record to value. N)rrrr!rHr r)r9rvalues r2rzMaskedRecords.__setitem__Es< dE2 dC !u5DJJt  !rlc h|jdkDrt|jjDcgc]}t ||c}Dcgc].}ddj |Dcgc] }t |c}d0}}}ddj |dSt|jjDcgc]}t ||c}Dcgc]+}dj |Dcgc] }t |c}-}}}ddj |dScc}wcc}wcc}}wcc}wcc}wcc}}w)z8 Calculates the string representation. r(r)[z, ])rJr$r r)r]joinr!)r9fsr+mstrs r2__str__zMaskedRecords.__str__Ns% 99q= TZZ=M=M"N74#3"NOQ!!4Q#a&!456a8QDQtyy'q) )!DJJ|D cgc] } t| } } |D cgc] } tjt| "} } t | |||||||j t } tt| | j_ ||| _ | Scc} wcc} w)a  Creates a mrecarray from a (flat) list of masked arrays. Parameters ---------- arraylist : sequence A list of (masked) arrays. Each element of the sequence is first converted to a masked array if needed. If a 2D array is passed as argument, it is processed line by line dtype : {None, dtype}, optional Data type descriptor. shape : {None, integer}, optional Number of records. If None, shape is defined from the shape of the first array in the list. formats : {None, sequence}, optional Sequence of formats for each individual field. If None, the formats will be autodetected by inspecting the fields and selecting the highest dtype possible. names : {None, sequence}, optional Sequence of the names of each field. fill_value : {None, sequence}, optional Sequence of data to be used as filling values. Notes ----- Lists of tuples should be preferred over lists of lists for faster processing. r r7rBr)rCrErD) rr( atleast_1dr recfromarraysrarr r$rr8rR) arraylistr r7rBr)rCrErDrRxdatalistmasklist_arrays r2rrs>%..q .H.8AB1 l1o.BHB 8!&eW!&vw%.015Y S(^,FLL& M/Bs B%Bc t|dd} t|trQt|tr#|j j t}| |j }|j}t||||||||j t} ||| _ | turtj| } t| j } | r| | j_nI| j"dk(r)| D cgc] } t%| c} | j_n| j'| | | | jdd| Scc} w)aN Creates a MaskedRecords from a list of records. Parameters ---------- reclist : sequence A list of records. Each element of the sequence is first converted to a masked array if needed. If a 2D array is passed as argument, it is processed line by line dtype : {None, dtype}, optional Data type descriptor. shape : {None,int}, optional Number of records. If None, ``shape`` is defined from the shape of the first array in the list. formats : {None, sequence}, optional Sequence of formats for each individual field. If None, the formats will be autodetected by inspecting the fields and selecting the highest dtype possible. names : {None, sequence}, optional Sequence of the names of each field. fill_value : {None, sequence}, optional Sequence of data to be used as filling values. mask : {nomask, sequence}, optional. External mask to apply on the data. Notes ----- Lists of tuples should be preferred over lists of lists for faster processing. rNrrr)r]rr rr rar rrecfromrecordsrrRrr(rrrr8rorrM)reclistr r7rBr)rCrErDrRrPrmrecmaskrecordlengthrYs r2rr"sD GWd +E'7# g{ +nn&++G4G =MME.." 'eW %f")Y @@DY $ 6zz$tzz? "DJJO YY!^156AuQx6DJJO   T "  1 K 7s7Ecg}tj|}|jdk(r|d}n|jdkDr td|D]6} t ||j tj t8|S#ttf$r t||j tj tnz#ttf$rh t||j tj tn0#ttf$r|j |j YnwxYwYnwxYwYwxYw)a. Tries to guess the dtypes of the str_ ndarray `arr`. Guesses by testing element-wise conversion. Returns a list of dtypes. The array is first converted to ndarray. If the array is 2D, the test is performed on the first line. An exception is raised if the file is 3D or more. rrrzThe array should be 2D at most!) r(rrorintr'r ryfloatcomplex)arrvartypesrs r2_guessvartypesrcsH **S/C xx1}!f A:;; + + F OOBHHSM *!+" OI& 1 1a0 * 77AJOOBHHW$56#I./OOCII./ 7 1sZ BE C)ED? D #)D? *D9 6D?8D9 9D?<E>D??EEct|dr|S t|}|jdddk7r|j dd|S|j t d#t$r}td|d|d}~wwxYw) z1 Opens the file handle of file `fname`. readlinezNo such file: ''Nrrz\xrzWow, binary file)hasattropenFileNotFoundErrorrseekcloser|)fnamerrs r2openfilers uj! C K zz|BQ5  q! GGI 0 11 C/% :;BCs A A>)A99A>) delimitorc ~|tjur+| tdtjdt d|}t |} |j}|d|j|j} | j|} t| dkDrnS|| }t|Dcgc]7}|d|k7r-t|dkDr|jj|9c}} | j\} } |j|t| d}nj|Dcgc]}tj |}}t|| k7r:d}|d z }tj|t|| fzdt| d}t#||Dcgc] \}}||f }}}|Dcgc]}t%j&|}}| j(|k(}t#| j(|||Dcgc]\}}}}t|||| }}}}}t+|| Scc}wcc}wcc}}wcc}wcc}}}}w) a Creates a mrecarray from data stored in the file `filename`. Parameters ---------- fname : {file name/handle} Handle of an opened file. delimiter : {None, string}, optional Alphanumeric character used to separate columns in the file. If None, any (group of) white spacestring(s) will be used. commentchar : {'#', string}, optional Alphanumeric character used to mark the start of a comment. missingchar : {'', string}, optional String indicating missing data, and used to create the masks. varnames : {None, sequence}, optional Sequence of the variable names. If None, a list will be created from the first non empty line of the file. vartypes : {None, sequence}, optional Sequence of the variables dtypes. If None, it will be estimated from the first non-commented line. Ultra simple: the varnames are in the header, one lineNz;fromtextfile() got multiple values for argument 'delimiter'zThe 'delimitor' keyword argument of numpy.ma.mrecords.fromtextfile() is deprecated since NumPy 1.22.0, use 'delimiter' instead.rr) stacklevelrrz&Attempting to %i dtypes for %i fields!z Reverting to default.)rPr rRr5)r(_NoValuerywarningswarnDeprecationWarningrrfindstripr"rrr7rrr r$rHdefault_fill_valueTr)r delimiter commentchar missingcharvarnamesrrftextline firstline _varnames _variablesrnfieldsvrXr/rr:mfillvrarYr1 _datalists r2rrsO4 #  *+ +  E)Q 8  UOE ~~0$))K01779 OOI. y>A    L"&q'["8SY] $zz|11)<LMJ##LQ KKM!*Q-0)12ABHHQK22 x=G #:C + +C MM#X 88Q G%jm4H$'x#: ;!Qq!f ;F ;08 91b##A& 9F 9\\[ (E%(uh%OQQ!aAqaaqQ?QIQ iv ..5L3< 9 Qs*s5;;// xx!Hxx ))lHNN-K,LLMHu{{H-Gkk  '')+ Wnenna(-1-+GX^^Igmm&:&:<&HIll=)G?11bgg,?@Iu{{I.Gkk  '')+ Wnenna(-1-+G\(+:mm**<8:GM N!+@+s'HH1'H!r)NNNNNFNN)N#NN))rnumpy.marrrrrrr r rHr numpyr(r r r r`numpy._core.recordsrrrrr_corerecords_byteorderconvcore_check_fill_value__all__r&r3r<rrrrrr rrrtrlr2r3s 11!!00GG--  <DN%KN%b H  ;?AE(V $4TUdf>B"J2(FH)-M/ kkM/`"rl