J/Ph2gdZddlZddlZddlmZddlmZddlm Z ddlm Z m Z m Z m Z mZddlmZddlmZdd lmZmZGd d Zd ZGd dZdZGddeZdZdZdZdZdZ dZ!Gddej"Z#dZ$dS)) DesignInfo FactorInfo SubtermInfo DesignMatrixN) PatsyError)atleast_2d_column_default) OrderedDict)repr_pretty_delegaterepr_pretty_implsafe_issubdtype no_picklingassert_no_pickling)linear_constraint)ContrastMatrix) ModelDescTermc(eZdZdZddZeZdZeZ dS)ra3A FactorInfo object is a simple class that provides some metadata about the role of a factor within a model. :attr:`DesignInfo.factor_infos` is a dictionary which maps factor objects to FactorInfo objects for each factor in the model. .. versionadded:: 0.4.0 Attributes: .. attribute:: factor The factor object being described. .. attribute:: type The type of the factor -- either the string ``"numerical"`` or the string ``"categorical"``. .. attribute:: state An opaque object which holds the state needed to evaluate this factor on new data (e.g., for prediction). See :meth:`factor_protocol.eval`. .. attribute:: num_columns For numerical factors, the number of columns this factor produces. For categorical factors, this attribute will always be ``None``. .. attribute:: categories For categorical factors, a tuple of the possible categories this factor takes on, in order. For numerical factors, this attribute will always be ``None``. Nch||_||_|jdvrtd|j||_|jdkr6t |t std|tdn-|jdksJ|tdt |}||_||_dS)N) numerical categoricalz:FactorInfo.type must be 'numerical' or 'categorical', not rz5For numerical factors, num_columns must be an integerz.For numerical factors, categories must be Nonerz1For categorical factors, num_columns must be None) factortype ValueErrorstate isinstanceinttuple num_columns categories)selfrrrrrs Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/patsy/design_info.py__init__zFactorInfo.__init__Os  98 8 8*:>))F  9 # #k3//  N% !TUUU&9 ----& !WXXXz**J&$c|rJGddt}d|jfd|jfd|fg}|jdkr|d|jfn|d|jft ||g|dS) NceZdZdZdS)-FactorInfo._repr_pretty_..FactorStatecdS)Nzr s r!__repr__z6FactorInfo._repr_pretty_..FactorState.__repr__ms''r#N)__name__ __module__ __qualname__r*r(r#r! FactorStater&ls# ( ( ( ( (r#r.rrrrrr)objectrrappendrrr )r pcycler.kwlists r! _repr_pretty_zFactorInfo._repr_pretty_is ( ( ( ( (& ( ( ( t{ # TY  kkmm $   9 # # MM=$*:; < < < < MM<9 : : :D"f-----r#NN r+r,r-__doc__r"r r*r4r __getstate__r(r#r!rr*sJ""H%%%%0$H....LLLr#rc tddddid}|jdksJ|jddiksJ|jdksJ|jdksJ|jJt |tddddid d g }|jdksJ|jddiksJ|jdksJ|jJ|jd ksJt |d dl}|ttddi|ttddi|ttddid|ttddidd|ttddi|ttddid|ttddid dS)Nasdfra rrzjr)r@rArz non-numerical)rr) rrrrrrreprpytestraisesr TypeError)fi1fi2rDs r!test_FactorInforIs V[3( C C CC :     9a 8{ " " " " ?b > ! ! ! III V]S!H#s L L LC :     9a 8} $ $ $ $ ? " " " >Z ' ' ' ' IIIMMM MM*j&/2FFF MM*j&+rBBB MM*j&+rvMVVV MMJ RQST MM)ZCCC MM*j&-QRMSSS MM)ZqMQQQQQr#c&eZdZdZdZeZdZeZ dS)raA SubtermInfo object is a simple metadata container describing a single primitive interaction and how it is coded in our design matrix. Our final design matrix is produced by coding each primitive interaction in order from left to right, and then stacking the resulting columns. For each :class:`Term`, we have one or more of these objects which describe how that term is encoded. :attr:`DesignInfo.term_codings` is a dictionary which maps term objects to lists of SubtermInfo objects. To code a primitive interaction, the following steps are performed: * Evaluate each factor on the provided data. * Encode each factor into one or more proto-columns. For numerical factors, these proto-columns are identical to whatever the factor evaluates to; for categorical factors, they are encoded using a specified contrast matrix. * Form all pairwise, elementwise products between proto-columns generated by different factors. (For example, if factor 1 generated proto-columns A and B, and factor 2 generated proto-columns C and D, then our final columns are ``A * C``, ``B * C``, ``A * D``, ``B * D``.) * The resulting columns are stored directly into the final design matrix. Sometimes multiple primitive interactions are needed to encode a single term; this occurs, for example, in the formula ``"1 + a:b"`` when ``a`` and ``b`` are categorical. See :ref:`formulas-building` for full details. .. versionadded:: 0.4.0 Attributes: .. attribute:: factors The factors which appear in this subterm's interaction. .. attribute:: contrast_matrices A dict mapping factor objects to :class:`ContrastMatrix` objects, describing how each categorical factor in this interaction is coded. .. attribute:: num_columns The number of design matrix columns which this interaction generates. ct||_t|}t|tst d|D]?\}}||vrt dt|tst d|@||_t|tst d||_ dS)Nzcontrast_matrices must be dictz+Unexpected factor in contrast_matrices dictzExpected a ContrastMatrix, not znum_columns must be an integer) rfactors frozensetrdictritemsrcontrast_matricesrr)r rLrPr factor_setrcontrast_matrixs r!r"zSubtermInfo.__init__sW~~ w'' +T22 ?=>> >'8'>'>'@'@   #FOZ'' !NOOOo~>>  j;J?L "3+s++ ?=>> >&r#c b|rJt||gd|jfd|jfd|jfgdS)NrLrPr)r rLrPrr r1r2s r!r4zSubtermInfo._repr_pretty_sX  DL)$d&<= 01  r#Nr6r(r#r!rrsE**X'''"$H    LLLr#rcfttjdddg}tddgd|id}|jdksJ|jd|iksJ|jdksJt|ddl}| ttd id | ttddgd d | ttddgd |id | ttddgdd id | ttddgid dS) Nr?r?[1][2]r;x)r;rYrr<r@g?) rnponesrrLrPrrCrDrErFr)cmsrDs r!test_SubtermInfor_s* % 8 8BS#Jb 1--A 9 " " " " 3) + + + + =A     GGGMMM MM)[!R333 MM*kC:q!<<< MM*kC:Ry!DDD MM*kC:QxCCC MM*kC:r3?????r#ceZdZdZddZeZdZedZ edZ edZ edZ ed Z d Zd Zd Zd ZeddZeZdS)rafA DesignInfo object holds metadata about a design matrix. This is the main object that Patsy uses to pass metadata about a design matrix to statistical libraries, in order to allow further downstream processing like intelligent tests, prediction on new data, etc. Usually encountered as the `.design_info` attribute on design matrices. Nc H tt|tt||_|du|dukrt d||_||_|jt|jtst dt|jtst d|j D]\}}t|tst d|t|tst dt|j}|D]R}t|tst d|||jst dSt}|jD]}||j|t|jkrt d|j D]F\} } t| t$st d | | | jkrt d G|j D]\}}|D]}d } t} |jD]} |j| } | jd kr | | jz} %| jd ksJ|j| j}|jdt| jkrt d| | | | |jd z} | t|jkrt d| |jkrt d|Rd|_|}dtt|D}tt|||_nt|_d}|j D]6\}}d}|D] }||jz } t;|||z|j|<||z }7|t|jkrt dtd|j D|_|jJ|jPt|jt|jksJd}|jD]:}| t|\}}}||ksJ|d ksJ|};|t|ksJ|j D]A\}}||jvr3|j|}|t;||d zkrt dBdS)NzFMust specify either both or neither of factor_infos= and term_codings=zfactor_infos should be a dictz#term_codings must be an OrderedDictzexpected a Term, not zterm_codings must contain listszexpected SubtermInfo, not zunexpected factors in subtermz3Provided Term objects and factor_infos do not matchz expected FactorInfo object, not zmismatched factor_info.factorr<rrrz&Mismatched contrast matrix for factor z:Mismatch between contrast_matrices and categorical factorszUnexpected num_columnsc4g|]}t||dzS)r<)slice).0is r! z'DesignInfo.__init__..es$HHH!eAq1uooHHHr#z>mismatch between column_names and columns coded by given termsc@g|]\}}||fSr(name)rdtermslice_s r!rfz'DesignInfo.__init__..vs)VVV>D&$))++v&VVVr#zterm/column name collision)!r ziprangelencolumn_name_indexesr factor_infos term_codingsrrNrOrlistsetrLr issupersetupdaterrrrrPmatrixshaperadd term_slicesterm_name_slicesrc column_namesvaluesindices)r r{rprqrjsubterms term_factorssubterm all_factorsr factor_infoexp_cols cat_factorsfir] term_namesslicesidx subterm_infos term_columns subterm_infocoveredrkstartstopstep column_nameindexs r!r"zDesignInfo.__init__sw#.  eC $5$566 7 7$ $   D ld&: ; ;2  )(   (d/66 B !@AAAd/== H !FGGG"&"3"9"9";"; J Jh!$--J$*$$%HIII!(D11H$%FGGG"4<00 'JJG%g{;;Y(jgg)WXXX'227?CCJ()HIIIJJ %%K) 1 1""4<0000c$"34444 L(,'8'>'>'@'@ F F# !+z::$*@K M[///$%DEEE0#'"3"9"9";"; C Ch'CCG H"%%%K")/ 4 4!.v67k11$6HH#%7m#;#;#;#;!(!:6!B!IB!x{c"-.@.@@@&0j7=v%@'"'"!"(OOF333$ 3HH"c'*C&D&DDD(6 7#666()ABBB7-C2  #D %JHHuS5F5F/G/GHHHF$/J0G0G$H$HD ! ! +}}D C'+'8'>'>'@'@ $ $#m $1==L L$<4CT2U V      r#c*t|jS)z%A list of the column names, in order.)rrror)s r!r{zDesignInfo.column_namessD,---r#c<|jdSt|jS)z1A list of :class:`Terms`, in order, or else None.N)ryrrr)s r!termszDesignInfo.termss#   #4D$%%%r#c*t|jS)zA list of terms, in order.)rrrzr)s r!rzDesignInfo.term_namessD)***r#cLtjtdd|S).. deprecated:: 0.4.0zThe DesignInfo.builder attribute is deprecated starting in patsy v0.4.0; distinct builder objects have been eliminated and design_info.builder is now just a long-winded way of writing 'design_info' (i.e. the .builder attribute just returns self)r? stacklevelwarningswarnDeprecationWarningr)s r!builderzDesignInfo.builders;       r#cLtjtdd|S)rzStarting in patsy v0.4.0, the DesignMatrixBuilder class has been merged into the DesignInfo class. So there's no need to use builder.design_info to access the DesignInfo; 'builder' already *is* a DesignInfo.r?rrr)s r! design_infozDesignInfo.design_infos;  -        r#ct|tr|Stjt |tjrt||dzS|j||jvr |j|S||jvr |j|S||jvr |j|}t||dzStd|d)a Locate a subset of design matrix columns, specified symbolically. A patsy design matrix has two levels of structure: the individual columns (which are named), and the :ref:`terms ` in the formula that generated those columns. This is a one-to-many relationship: a single term may span several columns. This method provides a user-friendly API for locating those columns. (While we talk about columns here, this is probably most useful for indexing into other arrays that are derived from the design matrix, such as regression coefficients or covariance matrices.) The `columns_specifier` argument can take a number of forms: * A term name * A column name * A :class:`Term` object * An integer giving a raw index * A raw slice object In all cases, a Python :func:`slice` object is returned, which can be used directly for indexing. Example:: y, X = dmatrices("y ~ a", demo_data("y", "a", nlevels=3)) betas = np.linalg.lstsq(X, y)[0] a_betas = betas[X.design_info.slice("a")] (If you want to look up a single individual column by name, use ``design_info.column_name_indexes[name]``.) r<Nzunknown column specified '') rrcr[ issubdtyperintegerryrzror)r columns_specifierrs r!rczDesignInfo.slicesB ' / / %$ $ =/00"* = = C*,=,ABB B   ',=AQ,Q,Q#$56 6  5 5 5():; ;  8 8 8*+<=CcAg&& &j.s EEE4499;;EEEr#N)rprq)rstrr from_formula lhs_termlistr rhs_termlistrqr{rzrrir getryrLrp) r which_termsdesc new_namestterm_name_to_termrjnew_column_namesnew_factor_infosnew_term_codings name_or_termr^fs r!subsetzDesignInfo.subsetRst k3 ' ' F)+66D  J !HIIIEE43DEEEK   $I  I IT.t/DQ/GHH i(( ( " ) 6 615!$))++..! ! *}}  + A A (,,\<HH$T* D$5a$88 ??A*.*;A*>$Q'')-):4)@ && -- r#columnct|drt|j|r|jSt|d}|jdkrt dt |dt|jd}t|dr,t|j tj s d |D}nfd |D}t|S) aFind or construct a DesignInfo appropriate for a given array_like. If the input `array_like` already has a ``.design_info`` attribute, then it will be returned. Otherwise, a new DesignInfo object will be constructed, using names either taken from the `array_like` (e.g., for a pandas DataFrame with named columns), or constructed using `default_column_prefix`. This is how :func:`dmatrix` (for example) creates a DesignInfo object if an arbitrary matrix is passed in. :arg array_like: An ndarray or pandas container. :arg default_column_prefix: If it's necessary to invent column names, then this will be used to construct them. :returns: a DesignInfo object rT)preserve_pandasr?z&design matrix can't have >2 dimensionscolumnsr<dtypec,g|]}t|Sr()r)rdobjs r!rfz)DesignInfo.from_array..s888CHH888r#cg|]}| Sr(r()rdredefault_column_prefixs r!rfz)DesignInfo.from_array..s$QQQA&;&;QQ?QQQr#)hasattrrrrndimrgetattrrmrwr rr[rr)cls array_likerarrrr{s ` r! from_arrayzDesignInfo.from_arrays$ :} - - **  "C3 3  *) )' DIII 8a<<EFF F#y% ! *=*=>> 7G $ $ R_W]BJ-W-W R88888LLQQQQQQQL,'''r#r5)r)r+r,r-r7r"r r*r4propertyr{rrrrrcrrr classmethodrr r8r(r#r!rr sBACACACACF$H   ..X.&&X& ++X+  X   X ,Q,Q,Q\1F1F1Ff!!!B[[[z((([(>LLLr#rcddl}Gddt}|d}|d}t|g}t|g}|t|did|t|did i}t |t |gidgf|t |gid gfg}t gd ||}|jgd ksJ|jddgksJ|j ||gksJ|j dd d dd ksJ|j tddtdd d ksJ|j |tdd|tdd iksJ|dksJ| d td d ksJ| dtdd ksJ| dtd d ksJ| dtd dksJ| dtddksJ| |tddksJ| dtdd ksJ| |tdd ksJ| td d td d ksJ|t |j dt#|t%|t gd}|jgdksJ|jgdksJ|j J|j dd d ddksJ|j tdd td d td dtdd dksJ|j J|dksJ| d td d ksJ| dtdd ksJ| dtd d ksJ| dtd dksJ| dtdd ksJt gddksJ|t&t gd ||t&t gd ||t&t gd t)|||t&t gd|||t&t gd|||t&t gd|||t&t gd ||t |} | |=|t&t gd|| |d!} t| g} t |} t | gid g| | <|t&t gd"|| t+|} d#| |<|t&t gd | |t+|}t| did||<|t&t gd |||t&t gd |t+|t |}|||d$<|t&t gd ||t |}t-||||<|t&t gd ||t |}d%||d<|t&t gd ||t |}t |gid ||d<|t&t gd ||| t| d&iddg'i}t | t | g| t/t1jd(d)d*gid gfg}|t&t d+d,g||t| |g}| t| d&iddg'|t|did i}t |t | |g| t/t1jd-d)d*g|t/t1jd-d)d*gid gfg}|t&t gd.||t |t | |gid gfg}|t&t4ft gd.||t |t | |g| t/t1jd/gd0id1gfg}|t&t gd2||dS)3NrceZdZdZdZdS)$test_DesignInfo.._MockFactorc||_dSN_name)r ris r!r"z-test_DesignInfo.._MockFactor.__init__s DJJJr#c|jSrrr)s r!riz)test_DesignInfo.._MockFactor.names : r#N)r+r,r-r"rir(r#r! _MockFactorrs2        r#rrYyrr>r<)x1x2x3rr?rZ)rYrzx + yrrrr:a1a2a3bza1 + a2 + a3 + brrrr)rr;rz 1 + a + b)rp)rq)rrry1y2)rrr)rrrr)rrrrr;)rrrrr;zwhat is this I don't evenzthis is a stringznot a SubtermInforrB)rr?rWrXza[1]za[2]rV) a[1]:x[1] a[2]:x[1] a[1]:x[2] a[2]:x[2])r?r)rWrXz[3])rrz a[3]:x[1]rrz a[3]:x[2])rDr/rrr rrr{rrrorzrcryrrErrCrrrrrNrrr[r\KeyError)rDrf_xf_yt_xt_yrprqditerm_codings_x_onlyf_at_aterm_codings_with_anot_factor_infosmismatch_factor_infosnot_term_codingsnon_list_term_codingsnon_subterm_term_codings bad_subtermfactor_codings_aterm_codings_a_bad_rowst_axfactor_codings_axterm_codings_ax_extra_cmterm_codings_ax_missing_cm%term_codings_ax_wrong_subterm_columnss r!test_DesignInfors MMMf +c  C +c  C u++C u++C Z["! < < < Z["! < < <L SE2q))* +cKr14M4M3N-OPL +++\< H HB ?555 5 5 5 5 =S#J & & & & 8Sz ! ! ! ! !AQaa%H%H H H H H a %1++"F"F F F F F >c5A;;U1a[[A A A A A ;;==G # # # # 88A;;%1++ % % % % 88D>>U1a[[ ( ( ( ( 88D>>U1a[[ ( ( ( ( 88D>>U1a[[ ( ( ( ( 88C==E!QKK ' ' ' ' 88C==E!QKK ' ' ' ' 88C==E!QKK ' ' ' ' 88C==E!QKK ' ' ' ' 88E!QKK E!QKK / / / / MM*bh/// HHHr +++ , ,B ?555 5 5 5 5 =333 3 3 3 3 8    !AQaa%H%H H H H H AqkkAqkkAqkk 1a[[ ##     > ! ! ! ;;==. . . . . 88A;;%1++ % % % % 88D>>U1a[[ ( ( ( ( 88D>>U1a[[ ( ( ( ( 88D>>U1a[[ ( ( ( ( 88C==E!QKK ' ' ' ' --- . . 7 7 9 9[ H H H H MMJ 7 7 7l MMJ 7 7 7l MM \  MM&&&  MMJ 2 2 2L, MMJ 7 7 7| MMJ 7 7 7| &l33C  MMJ 2 2 2LBU +c  C u++C%l33 +SE2q 9 9: MM$$$ L))7S MMJ 7 7 79I<!..!+Cba!P!P!P# MM  MM \ #<00+7+<'( MMJ 7 7 7GW( 55!&|C'8!9!9# MM  +<88':S!!$ MM l++K&ub!44La MMJ 7 7 7{ Z]BD$< H H H*^BGFOOeU^%T%TUWX     MM   c  D Z]BD$< H H H Z["! < < < +c %!P!P%!P!P      " MM<<< "-tk3*bRS6T6T5U.V-W!X!X MM X<<<" -8c nRWV__>S>S>STTU   --) MMVVV- r#cVtgd}|jdgksJtddgddgddgg}|jddgksJtgdd }|jd gksJtddgddgddggd }|jd d gksJtgd|}t||usJd |_t|}|jdgksJd dl}|ttjtj dd dl m }|rd dl }t| ddggddg} | jddgksJt| ddggd dg} | jddgksJ| ddgg} | | _t| | usJdSdS)N)r<r?rcolumn0r<r?rrZcolumn1rYrx0rr:r)r?r?r?) have_pandasr;r)rr=column10)rrr{rrrDrErr[r\ patsy.utilr pandas DataFrame) rdi2di3di4mdi_weirdrDr r di5di6dfs r!test_DesignInfo_from_arrayrsx   yyy ) )B ?yk ) ) ) )   !Q!Q!Q 8 9 9C   95 5 5 5 5      E EC  v % % % %   !Q!Q!Q 8PS  T TC  d| + + + +YYY$$A   # #s * * * *AM$$Q''H  YK / / / /MMM MM*j3RWY5G5GHHH&&&&&& 0 ##F$4$4q!fXSz$4$R$RSSC:----##F$4$4q!fX2w$4$O$OPPIz#:::::   1vh ' '$$R((C//// 0 00/r#ctgd}|ddg}|jgdksJtj|jgdgdgksJtj|jdgdggksJdS)Nrz2 * a1 = b + 1r)r?rr)rrr<rr<r)rrvariable_namesr[allcoefs constants)rcons r!!test_DesignInfo_linear_constraintr s +++ , ,B   0$7 8 8C  !8!8!8 8 8 8 8 6#) |||<< = ==== 6#-QC!:- . ......r#c`tddgfd}|d|ddS)Nrrc tjd5}tjdt|usJ dddn #1swxYwYt |dksJ|djt usJdS)NT)recordalwaysr<r)rcatch_warnings simplefilterrrncategoryr)attrwds r!checkz4test_DesignInfo_deprecated_attributes..checks  $D 1 1 1 )Q  !( + + +1d##q((((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )1vv{{{{t} 2222222s)A  AArr)r)r*r)s @r!%test_DesignInfo_deprecated_attributesr+sUD$<  A33333 E) E-r#cdt|zdz|jdksJtdtjfd|Dt }tjfd|D}tj|d}d tj|rK tjfd ||Dr|D]}||dd ||<n d krd nnJ|S)Nz%.rr<z +-0123456789.cg|]}|fz Sr(r()rdrY format_strs r!rfz(_format_float_column..,s 888qqd*888r#rcBg|]}|od|vS).)rt)rdcol_strsimple_float_charss r!rfz(_format_float_column..0s@     ) )' 2 2 Esg~   r#r0Tc:g|]}|Sr()endswith)rdr^ strip_chars r!rfz(_format_float_column..9s%FFF!qzz*--FFFr#rr1) rrrsr[arrayr/nonzeroanyr) precisioncolcol_strsmask mask_idxesrr.r3r7s @@@r!_format_float_columnr@'sCI&,J 8q====_--x8888C888GGGH 8    #     D D!!!$JJ vd||  vFFFFx~FFFGG %77C$,SM#2#$6HSMM7$$!$JJ  Or#c$d}dtjfz}|ddddtjgddd |g|ddd dtjgd d d |g|ddd dtjgd d d |g|ddd dtjgddd|gdS)Nct|tj|}t||tj||sJdSr)r@r[asarrayprint array_equal)r;numbersexpectedgots r!rz$test__format_float_column..tEsI"9bj.A.ABB c8~c8,,,,,,,r#z%.3frr<gx#@gH@z1.000z2.123z2.124r?r23gqh?rZz1.0001z2.0000z3.0000)r[nan)r nan_strings r!test__format_float_columnrMDs---26)#JAa!VVRV $w*&MNNNAa!Q26 S#sJ7888Aa&!Q #sC!<===Aa&!Q (Hh !KLLLLLr#c(eZdZdZddZeZdZeZ dS)ra1A simple numpy array subclass that carries design matrix metadata. .. attribute:: design_info A :class:`DesignInfo` object containing metadata about this design matrix. This class also defines a fancy __repr__ method with labeled columns. Otherwise it is identical to a regular numpy ndarray. .. warning:: You should never check for this class using :func:`isinstance`. Limitations of the numpy API mean that it is impossible to prevent the creation of numpy arrays that have type DesignMatrix, but that are not actually design matrices (and such objects will behave like regular ndarrays in every way). Instead, check for the presence of a ``.design_info`` attribute -- this will be present only on "real" DesignMatrix objects. Nrct|trt|dr|St||}t |jtjr.tj |t|}|j dkrtd|j dksJ|t||}t|j|jdkr3tdt|jd|jdd ||_t |jtjstd |S) aCreate a DesignMatrix, or cast an existing matrix to a DesignMatrix. A call like:: DesignMatrix(my_array) will convert an arbitrary array_like object into a DesignMatrix. The return from this function is guaranteed to be a two-dimensional ndarray with a real-valued floating point dtype, and a ``.design_info`` attribute which matches its shape. If the `design_info` argument is not given, then one is created via :meth:`DesignInfo.from_array` using the given `default_column_prefix`. Depending on the input array, it is possible this will pass through its input unchanged, or create a view. rr/r?zDesignMatrix must be 2dNr<z4wrong number of column names for design matrix (got z , wanted )z0design matrix must be real-valued floating point)rrrrviewr rr[rrCfloatrrrrrnr{rwrfloating)r input_arrayrrr s r!__new__zDesignMatrix.__new__jsX0 k< 0 0 W 6 6   (55::3?? 4:rz 2 2 ;:d%00055c::D 9q==677 7yA~~~~  $//6KLLK {' ( (DJqM 9 9*),[-E)F)F)F)F ST W 'tz2;77 QOPP P r#cLt|ds)|tj|dS|rJd}d}d}d}d|jj}d|D}|||jddz zztj|z} | |krtj|d|ddffdt|jdD} d fd | D} d t|| D} |||jddz zztj| z} | |k}nd }| |d |j| dd|j zz|rd|z}|gtt| zD]d}dt| |D}||||dd|j zze||jdkrG|d|jd|z d|dd|j zzn| dd| dd|j zz|||dd| dd|j zz| dd| dd|j zz|jjD]\}}|jdkr| d|||j|jz dkr d|j}nd|jd|j}|d|d|dd|r|jd|kr5| dd|j zz|d||ddS)NrNr?rc,g|]}t|Sr(rn)rdris r!rfz.DesignMatrix._repr_pretty_..s:::Dc$ii:::r#r<cDg|]}tdd|fSr)r@)rdre PRECISIONprintable_parts r!rfz.DesignMatrix._repr_pretty_..s@%Yqqq!t0DEEr#cl|jdksJ|jdsdStd|DS)Nr<rc,g|]}t|Sr(rZ)rdr^s r!rfzADesignMatrix._repr_pretty_..max_width..s4441A444r#)rrwmax)r<s r! max_widthz-DesignMatrix._repr_pretty_..max_widthsAx1}}}}y|6144444555r#c&g|] }|Sr(r()rdr<ras r!rfz.DesignMatrix._repr_pretty_..s! J J JC3 J J Jr#c4g|]\}}t||Sr()r`)rd name_width num_widths r!rfz.DesignMatrix._repr_pretty_..s6+ZJ **r#FzDesignMatrix with shape   c>g|]\}}||Sr()rjust)rdwidthcells r!rfz.DesignMatrix._repr_pretty_..s(XXX}tE**XXXr#r[z rows omitted]zColumns:zTerms:z, zcolumn zcolumns :z (rPz-(to view full data, use np.asarray(this_obj)))rprettyr[rCrr{rwsumrmrl begin_group breakable indentationrrtextr end_grouprzrOrr)r r1r2MAX_TOTAL_WIDTHSEPINDENTMAX_ROWSrcolumn_name_widthsmin_total_widthformatted_colscolumn_num_widths column_widths total_width print_numbersseprowcells term_namespancoltextr\rar]s @@@r!r4zDesignMatrix._repr_pretty_st]++  HHRZ%% & & & F  -::E::: SDJqMA-. .8J1K1K K  o - -Z--ixil;Ntz!}--N  6 6 6!K J J J> J J J /2&(900M !3$*Q-!*;#<&:!;!;; 3 3XXMSV@W@WXXXsxx'''tcAM112222$*Q-''djmh.F.F.FHIIItcAM11222 MM!Z ( ( ( KKsQ]22 3 3 3 HHUOOO KK2    KKsQ]22 3 3 3 a""" D3..///#/@FFHH ) )OItzQ D!!! HHY   y4:%***)-5+.2ZZZC FFFggg' ( ( ( ( Ar D 1  8 8 KKsQ]22 3 3 3 FFB C C C FBr#)Nr) r+r,r-r7rUr r*r4r __reduce__r(r#r!rrTsN*----^$HW W W zJJJr#rcddl}tgd}tgdg|}|jjgdksJtdg}|t tgdg|tgdg}|jjgdksJtgd}|jdksJ|t tdgggt|}||usJt|}||usJtgdgd }|jjgd ksJt||t tgd |t tgd |t tdd tgt|tttj dtttj ddzt|ddddftttj ddtttjtjgtttjdddgtttjdtttjdtttjddS)Nrr) r)rrcolumn2column3)rZr<r<rYr)r rrr)r<r?y@)r;rcr?dg@)r<rg@xDg4@)r<r)rr<)rr)rDrrrr{rErrwdiagonalrr/rCr[arangereshaperKinfzeros) rDrmmbad_dimm2mm3mm4mm5mm6s r!test_design_matrixrsMMM +++ , ,B '''(" - -B > &*A*A*A A A A A   F MM*l-=-=-=,>GGG ((() * *C ? '+W+W+W W W W W ''' ( (C 9     MM*lqcUG444 r  C "9999 r{{}} % %C b==== ((() E E EC ? '+C+C+C C C C Cs MM*lJJJ777 MM*lOOO<<< MM*lQ688,<=== HHHbinn % %&&&binns* + +,,,ABBEOOObinn,,X66 7 7888rvrv& ' '(((rvq$- . .///bhv&& ' '(((bhv&& ' '(((bhv&& ' '(((((r#)%__all__rnumpyr[patsyrr r patsy.compatr r r r r rpatsy.constraintrpatsy.contrastsr patsy.descrrrrIrr_r/rrrrr+r@rMndarrayrrr(r#r!rsL* F E E000000$$$$$$/.....******&&&&&&&&VVVVVVVVr!R!R!RHMMMMMMMM`@@@&EEEEEEEEPSSSl"0"0"0J///   $: M M M ddddd2:dddN/)/)/)/)/)r#