k{a8ldZddlZddlmZddlmZmZddlmZm Z dZ dZ dZ Gd d Z d Zd Zy) 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.0c|tt|ni}|tt|ni}t|j}t|j}||zr t 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. z"Can not compose overlapping cycles)nextitersetkeys ValueError)leftrightl_peekr_peekl_keyr_keys (/usr/lib/python3/dist-packages/cycler.py _process_keysr4sk"&!1T$t* rF"'"3T$u+ F  E  E u}=>> 5=cT|j|jk7rJtdj|j|jz|j|jz |j|jt 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_onec3HK|]}t|||zywN_cycler).0k_l_rs r zconcat..es$Ea2a52a5=1Es")r r formatby_keyrr)rrr r!s @@rconcatr%Ks$ yyEJJ44:F"&))ejj"8&*ii%**&<5;5>? ? B B #E499E FFrceZdZdZdZddZdZedZdZ e dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZeZdZeZy)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. ct|Sr)rselfs r__call__zCycler.__call__s T{rNc<t|tr1t|j|j|j|_n2|)|Dcgc]}t j |c}|_nd|_t|tr1t|j|j|j|_n2|)|Dcgc]}t j |c}|_nd|_t |j|j|_||_ycc}wcc}w)zf Semi-private init. Do not use this directly, use `cycler` function instead. N) isinstancer'_left_right_opcopyr_keys)r*rropvs r__init__zCycler.__init__s dF # DKKBDJ  1551$))A,5DJDJ eV $ ellEIIFDK  277A499Q<7DKDK"4::t{{; 68s D;Dc||jvSr)r2)r*rs r __contains__zCycler.__contains__sDJJrc,t|jS)z!The keys this Cycler knows about.)r r2r)s rr z Cycler.keyss4::rcf||k(ry||jvrtdj||||jvrtdj|||jj ||jj ||j 5||j jvr|j j||yt|jtr|jj||y|jDcgc] }|||i c}|_ ycc}w)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 key) r2r r#KeyErrorremoverr/r change_keyr-r.r')r*r:r;entrys rr>zCycler.change_keys #:  $** HCS)  djj O"FsF46 6 # s ;; "sdkk.>.>'> KK " "3 , F + JJ ! !#s + :>D3c +DDJDsD.cZ|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|]}|i ywr)rr4labels rr"z$Cycler._from_iter..s1%1s)listr.r2)clsrCitrrets ` r _from_iterzCycler._from_iters.($i1S11 G  rcttr7|j}ttfd|j DSt d)Nc3BK|]\}}t||ywrr)rrr4keys rr"z%Cycler.__getitem__..s Mtq!1S6 2Mz+Can only use slices with Cycler.__getitem__)r-slicer$rritemsr )r*rKtranss ` r __getitem__zCycler.__getitem__s= c5 !KKME#Mu{{}MN NJK Krc#K|j|jD]}t|y|j|j|jD]-\}}i}|j ||j ||/ywr)r/r.dictr0update)r*rabouts r__iter__zCycler.__iter__su ;;   !4j  !T[[9 1 1  1   sBBct|t|k7r#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__sP t9E "$$'I;eCJ<AB BdE3''rcttrt|tSttr7|j }t t fd|jDStS)z Outer product of two cyclers (`itertools.product`) or integer multiplication. Parameters ---------- other : Cycler or int c3BK|]\}}t||zywrr)rrr4r\s rr"z!Cycler.__mul__..s N11U7 3NrL) r-r'rintr$rrrNNotImplemented)r*r\rOs ` r__mul__zCycler.__mul__sR eV $$w/ / s #KKME#N NO O! !rc ||zSrrBr[s r__rmul__zCycler.__rmul__s e|rctttti}|jt |j St |j }t |j}||j||Sr)rZminrrr/rYr.r0)r*op_dictl_lenr_lens r__len__zCycler.__len__s[Wc* ;; tzz? "DJJDKK  wtxx ..rc t|ts tdtj|}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) r-r' TypeErrorr1rr2r.rZr0r/r*r\old_selfs r__iadd__zCycler.__iadd__#sd%(@A A99T?"8U3  U[[%,, B  rc t|ts tdtj|}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) r-r'rlr1rr2r.rr0r/rms r__imul__zCycler.__imul__5sd%(@A A99T?"8U3  U[[%,, B  rct|t|k7ry|j|jz rytdt||DS)NFc3,K|] \}}||k(ywrrB)rrTrUs rr"z Cycler.__eq__..Ls7da167s)rYr allrZr[s r__eq__z Cycler.__eq__Gs@ t9E " 99uzz !7c$&6777rc||k( SrrBr[s r__ne__z Cycler.__ne__NsEM""rc2tdtdi}|j7|jj t fd|D}dd|dS|j |jd}d}|j|j||j S) N+*c3(K|] }| ywrrBrr4labs rr"z"Cycler.__repr__..Ws,!qv,zcycler(z, )?z({left!r} {op} {right!r}))rr3r) rZrr/r poprDgetr0r#r.)r*op_maprFr3msgr}s @r__repr__zCycler.__repr__SssGS) ;; ))--/C,t,,CSG2cWA. .DHHc*B-C::4::"DKK:H Hrcd}t|jt}|D] }|d|dz } t|D]}|dz }|D]}|d||dz }|dz }!|d z }|S) Nz)rKzzzzz
z
z
)sortedr reprr )r*output sorted_keysrKdrs r _repr_html_zCycler._repr_html_^sTYYD1  *C SG5) )F *d A f F  /D1.. / g F   * rc|j}|Dcic] }|t}}|D] }|D]}||j||"|Scc}w)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. )r rDappend)r*r rrVrs rr$z Cycler.by_keylsd*yy"&'Qq$&y'' $A $A ad# $ $ (sA cl|j}ttd|jDS)z Simplify the cycler into a sum (but no products) of cyclers. Returns ------- simple : Cycler c3:K|]\}}t||ywrrrrr4s rr"z"Cycler.simplify..sDdaGAqMD)r$rrrN)r*rOs rsimplifyzCycler.simplifys( cDekkmDEEr)NN)__name__ __module__ __qualname____doc__r+r5r7propertyr r> classmethodrHrPrWr]rbrdrjrorqrurw__hash__rrr$ _transposerr%rBrrr'r'hs66"EH0L  (""/$$8#H I <JF Frr'cV|r |r tdt|dk(r,t|dts tdt|dSt|dk(rt |St|dkDr td|r%t t d|jDStd) 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||ywrrrs rr"zcycler..sEdaGAqMErz4Must have at least a positional OR keyword arguments)rlrYr-r'rrrrN)argskwargss rcyclerrsL 12 2 4yA~$q'6*45 5d1g Ta~ TQNO OcEfllnEFF J KKrct|trB|j}t|dk7r d}t ||j fd|D}tj ||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|] }| ywrrBr|s rr"z_cycler..s#!qv#r~)r-r'r rYr rrH)rCrFr rr}s @rrrs\ #vxx t9>FCS/ !hhj$s#   UC ((r)rr1 functoolsr itertoolsrroperatorrr __version__rr%r'rrrBrrrsB(V $ .G:ttn 8Lv)r