a8tdZddlZddlmZddlmZmZddlmZm Z dZ dZ dZ Gd d Z d Zd ZdS) a Cycler ====== Cycling through combinations of values, producing dictionaries. You can add cyclers:: from cycler import cycler cc = (cycler(color=list('rgb')) + cycler(linestyle=['-', '--', '-.'])) for d in cc: print(d) Results in:: {'color': 'r', 'linestyle': '-'} {'color': 'g', 'linestyle': '--'} {'color': 'b', 'linestyle': '-.'} You can multiply cyclers:: from cycler import cycler cc = (cycler(color=list('rgb')) * cycler(linestyle=['-', '--', '-.'])) for d in cc: print(d) Results in:: {'color': 'r', 'linestyle': '-'} {'color': 'r', 'linestyle': '--'} {'color': 'r', 'linestyle': '-.'} {'color': 'g', 'linestyle': '-'} {'color': 'g', 'linestyle': '--'} {'color': 'g', 'linestyle': '-.'} {'color': 'b', 'linestyle': '-'} {'color': 'b', 'linestyle': '--'} {'color': 'b', 'linestyle': '-.'} N)reduce)productcycle)muladdz0.10.0c8|tt|ni}|tt|ni}t|}t|}||zrt d||zS)a Helper function to compose cycler keys. Parameters ---------- left, right : iterable of dictionaries or None The cyclers to be composed. Returns ------- keys : set The keys in the composition of the two cyclers. Nz"Can not compose overlapping cycles)nextitersetkeys ValueError)leftrightl_peekr_peekl_keyr_keys &lib/python3.11/site-packages/cycler.py _process_keysr4s"&!1T$t**   rF"'"3T$u++   F   E   E u}?=>>> 5=c@|j|jkr>td|j|jz|j|jz ||t t fd|jDS)aS Concatenate `Cycler`\s, as if chained using `itertools.chain`. The keys must match exactly. Examples -------- >>> num = cycler('a', range(3)) >>> let = cycler('a', 'abc') >>> num.concat(let) cycler('a', [0, 1, 2, 'a', 'b', 'c']) Returns ------- `Cycler` The concatenated cycler. zBKeys do not match: Intersection: {both!r} Disjoint: {just_one!r})bothjust_onec3VK|]#}t|||zV$dSN_cycler).0k_l_rs r zconcat..es9EEa2a52a5=11EEEEEEr)r r formatby_keyrr)rrr r!s @@rconcatr%Ks$ yEJ44:F"&)ej"8&*i%*&<5;5>5>?? ? B B #EEEEE49EEE F FFrceZdZdZdZddZdZedZdZ e dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZeZdZeZdS)Cyclera Composable cycles. This class has compositions methods: ``+`` for 'inner' products (zip) ``+=`` in-place ``+`` ``*`` for outer products (`itertools.product`) and integer multiplication ``*=`` in-place ``*`` and supports basic slicing via ``[]``. Parameters ---------- left, right : Cycler or None The 'left' and 'right' cyclers. op : func or None Function which composes the 'left' and 'right' cyclers. c t|Sr)rselfs r__call__zCycler.__call__sT{{rNct|tr&t|j|j|j|_n|d|D|_nd|_t|tr&t|j|j|j|_n|d|D|_nd|_t |j|j|_||_dS)zf Semi-private init. Do not use this directly, use `cycler` function instead. Nc6g|]}tj|Scopyrvs r z#Cycler.__init__..s 5551$)A,,555rc6g|]}tj|Sr.r/r1s rr3z#Cycler.__init__..s 777A49Q<<777r) isinstancer'_left_right_opr_keys)r*rrops r__init__zCycler.__init__s dF # #  DKBBDJJ  65555DJJDJ eV $ $  elEIFFDKK  87777DKKDK"4:t{;; rc||jvSr)r9)r*rs r __contains__zCycler.__contains__sDJrc*t|jS)z!The keys this Cycler knows about.)r r9r)s rr z Cycler.keyss4:rc:krdS|jvr$td|jvr$td|j|j|j+|jjvr|jdSt|j tr|j dSfd|j D|_ dS)a Change a key in this cycler to a new name. Modification is performed in-place. Does nothing if the old key is the same as the new key. Raises a ValueError if the new key is already a key. Raises a KeyError if the old key isn't a key. Nz6Can't replace {old} with {new}, {new} is already a key)oldnewz2Can't replace {old} with {new}, {old} is not a keyc$g|] }|i Sr.r.)rentryrAr@s rr3z%Cycler.change_key..s"DDD3c +DDDr) r9r r#KeyErrorremoverr7r change_keyr5r6r')r*r@rAs ``rrFzCycler.change_keys9 #:: F $*  HCS))  dj O"FsF4466 6 # s ; "sdk.>'>'> K " "3 , , , , , F + + E J ! !#s + + + + + EDDDDDDDDJJJrcn|d}tfd|D|_h|_|S)a Class method to create 'base' Cycler objects that do not have a 'right' or 'op' and for which the 'left' object is not another Cycler. Parameters ---------- label : str The property key. itr : iterable Finite length iterable of the property values. Returns ------- `Cycler` New 'base' cycler. Nc3 K|]}|iV dSrr.)rr2labels rr"z$Cycler._from_iter..s'11%111111r)listr6r9)clsrIitrrets ` r _from_iterzCycler._from_itersD(c$ii1111S11111 G  rcttrG|}ttfd|DSt d)Nc3JK|]\}}t||VdSrr)rrr2keys rr"z%Cycler.__getitem__..s5MMtq!1S6 2 2MMMMMMrz+Can only use slices with Cycler.__getitem__)r5slicer$rritemsr )r*rQtranss ` r __getitem__zCycler.__getitem__s_ c5 ! ! LKKMME#MMMMu{{}}MMMNN NJKK Krc#K|j|jD]}t|VdS||j|jD]5\}}i}|||||V6dSr)r7r6dictr8update)r*rabouts r__iter__zCycler.__iter__s ;   ! !4jj     ! !T[99  1 1  1    rct|t|kr/tdt|dt|t||tS)z Pair-wise combine two equal length cyclers (zip). Parameters ---------- other : Cycler z&Can only add equal length cycles, not z and )lenr r'zipr*others r__add__zCycler.__add__sr t99E " "A$'IIAA47JJAABB BdE3'''rc ttrt|tSttrG|}t t fd|DStS)z Outer product of two cyclers (`itertools.product`) or integer multiplication. Parameters ---------- other : Cycler or int c3DK|]\}}t||zVdSrr)rrr2ras rr"z!Cycler.__mul__..s5NN11U7 3 3NNNNNNr) r5r'rintr$rrrSNotImplemented)r*rarTs ` r__mul__zCycler.__mul__s{ eV $ $ "$w// / s # # "KKMME#NNNN NNNOO O! !rc ||zSrr.r`s r__rmul__zCycler.__rmul__s e|rctttti}|jt |jSt |j}t |j}||j||Sr)r_minrrr7r^r6r8)r*op_dictl_lenr_lens r__len__zCycler.__len__s[Wc* ; tz?? "DJDK   wtx ...rct|tstdtj|}t |||_||_t|_t|j|j |j|_ |S)z In-place pair-wise combine two equal length cyclers (zip). Parameters ---------- other : Cycler z"Cannot += with a non-Cycler object) r5r' TypeErrorr0rr9r6r_r8r7r*raold_selfs r__iadd__zCycler.__iadd__#sp%(( B@AA A9T??"8U33  U[%, BB  rct|tstdtj|}t |||_||_t|_t|j|j |j|_ |S)z In-place outer product of two cyclers (`itertools.product`). Parameters ---------- other : Cycler z"Cannot *= with a non-Cycler object) r5r'rqr0rr9r6rr8r7rrs r__imul__zCycler.__imul__5sp%(( B@AA A9T??"8U33  U[%, BB  rct|t|krdS|j|jz rdStdt||DS)NFc3(K|] \}}||kVdSrr.)rrYrZs rr"z Cycler.__eq__..Ls*77da16777777r)r^r allr_r`s r__eq__z Cycler.__eq__Gs[ t99E " "5 9uz ! 577c$&6&6777777rc||k Srr.r`s r__ne__z Cycler.__ne__NsEM""rc*tdtdi}|j=|jt fd|D}dd|dS||jd}d}||j ||j S) N+*c3(K|] }|V dSrr.rr2labs rr"z"Cycler.__repr__..Ws',,!qv,,,,,,rzcycler(z, )?z({left!r} {op} {right!r}))rr:r) r_rr7r poprJgetr8r#r6)r*op_maprLr:msgrs @r__repr__zCycler.__repr__SssGS) ; )--//C,,,,t,,,,,C.S..c... .DHc**B-C::4:"DK:HH Hrcd}t|jt}|D] }|d|dz } t|D] }|dz }|D]}|d||dz }|dz }!|d z }|S) Nz)rQzzzzz
z
z
)sortedr reprr )r*output sorted_keysrQdrs r _repr_html_zCycler._repr_html_^sTYD111  * *C )S))) )FFd  A f F  / /.1.... g FF* rc|j}d|D}|D](}|D]#}||||$)|S)a Values by key. This returns the transposed values of the cycler. Iterating over a `Cycler` yields dicts with a single value for each key, this method returns a `dict` of `list` which are the values for the given key. The returned value can be used to create an equivalent `Cycler` using only `+`. Returns ------- transpose : dict dict of lists of the values for each key. c,i|]}|tSr.)rJ)rrs r z!Cycler.by_key..s'''Qq$&&'''r)r append)r*r r[rrs rr$z Cycler.by_keylsc*y''$''' $ $A $ $A ad#### $ rc|}ttd|DS)z Simplify the cycler into a sum (but no products) of cyclers. Returns ------- simple : Cycler c3<K|]\}}t||VdSrrrrr2s rr"z"Cycler.simplify..s.DDdaGAqMMDDDDDDr)r$rrrS)r*rTs rsimplifyzCycler.simplifys8 cDDekkmmDDDEEEr)NN)__name__ __module__ __qualname____doc__r+r;r=propertyr rF classmethodrNrUr\rbrgrirortrvrzr|__hash__rrr$ _transposerr%r.rrr'r'hs66X"E"E"EH[0LLL    ( ( (""""///$$888###H I I I   <JFFF FFFrr'c|r|rtdt|dkr?t|dtstdt|dSt|dkr t |St|dkrtd|r1t t d|DStd) a Create a new `Cycler` object from a single positional argument, a pair of positional arguments, or the combination of keyword arguments. cycler(arg) cycler(label1=itr1[, label2=iter2[, ...]]) cycler(label, itr) Form 1 simply copies a given `Cycler` object. Form 2 composes a `Cycler` as an inner product of the pairs of keyword arguments. In other words, all of the iterables are cycled simultaneously, as if through zip(). Form 3 creates a `Cycler` from a label and an iterable. This is useful for when the label cannot be a keyword argument (e.g., an integer or a name that has a space in it). Parameters ---------- arg : Cycler Copy constructor for Cycler (does a shallow copy of iterables). label : name The property key. In the 2-arg form of the function, the label can be any hashable object. In the keyword argument form of the function, it must be a valid python identifier. itr : iterable Finite length iterable of the property values. Can be a single-property `Cycler` that would be like a key change, but as a shallow copy. Returns ------- cycler : Cycler New `Cycler` for the given property zBcyl() can only accept positional OR keyword arguments -- not both.rzDIf only one positional argument given, it must be a Cycler instance.zdOnly a single Cycler can be accepted as the lone positional argument. Use keyword arguments instead.c3<K|]\}}t||VdSrrrs rr"zcycler..s.EEdaGAqMMEEEEEErz4Must have at least a positional OR keyword arguments)rqr^r5r'rrrrS)argskwargss rcyclerrsL 22122 2 4yyA~~$q'6** 5455 5d1g Ta~ TQNOO OGcEEfllnnEEEFFF J K KKrct|trM|j}t|dkrd}t ||fd|D}t||S)aD Create a new `Cycler` object from a property name and iterable of values. Parameters ---------- label : hashable The property key. itr : iterable Finite length iterable of the property values. Returns ------- cycler : Cycler New `Cycler` for the given property rz2Can not create Cycler from a multi-property Cyclerc3(K|] }|V dSrr.rs rr"z_cycler..s'##!qv######r)r5r'r r^r rrN)rIrLr rrs @rrrs| #v $x t99>>FCS// !hhjj$###s###   UC ( ((r)rr0 functoolsr itertoolsrroperatorrr __version__rr%r'rrr.rrrs((V $$$$$$$$ .GGG:ttttttttn 8L8L8Lv)))))r