a амGd'd(Дd(e.e=ГZ?d)S)*zq Finite Discrete Random Variables Module See Also ======== sympy.stats.frv_types sympy.stats.rv sympy.stats.crv щ)┌product)┌Sum)┌Basic)┌cacheit)┌Lambda)┌Mul)┌I┌nanй┌Eq)┌S)┌Dummy┌Symbolй┌sympifyй┌exp)┌ Piecewise)┌And┌Or)┌Intersection)┌Dict)┌Logic)┌ Relational)┌_sympifyй┌ FiniteSet)┌RandomDomain┌ ProductDomain┌ConditionalDomain┌PSpace┌IndependentProductPSpace┌SinglePSpace┌random_symbols┌sumsets┌rv_subs┌NamedArgsMixin┌Density┌Distributionc@s$eZdZdZddДZeddДГZdS)┌ FiniteDensityz' A domain with Finite Density. cCs t|Г}||vr||SdSdS)zЦ Make instance of a class callable. If item belongs to current instance of a class, return it. Otherwise, return 0. rNr)┌self┌itemйr,·_/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/stats/frv.py┌__call__(szFiniteDensity.__call__cCst|ГS)z, Return item as dictionary. )┌dictйr*r,r,r-r/6szFiniteDensity.dictN)┌__name__┌ __module__┌__qualname__┌__doc__r.┌propertyr/r,r,r,r-r)$sr)c@sPeZdZdZdZeddДГZeddДГZeddДГZd d ДZ ddДZ d dДZdS)┌FiniteDomainzS A domain with discrete finite support Represented using a FiniteSet. TcCstddД|jDГГS)Ncss|]\}}|VqdSйNr,й┌.0┌sym┌valr,r,r-┌ Gєz'FiniteDomain.symbols..йr┌elementsr0r,r,r-┌symbolsEszFiniteDomain.symbolscCs |jdSйNrй┌argsr0r,r,r-r?IszFiniteDomain.elementscCstddД|jDГОS)NcSsg|]}tt|ГГСqSr,)rr/)r9┌elr,r,r-┌ Or=z%FiniteDomain.dict..r>r0r,r,r-r/MszFiniteDomain.dictcCs ||jvSr7)r?йr*┌otherr,r,r-┌__contains__QszFiniteDomain.__contains__cCs |jабSr7)r?┌__iter__r0r,r,r-rITszFiniteDomain.__iter__cCstddД|DГОS)NcSsg|]}tddД|DГОСqS)cSsg|]\}}t||ГСqSr,r r8r,r,r-rEXr=z6FiniteDomain.as_boolean...)r)r9r+r,r,r-rEXr=z+FiniteDomain.as_boolean..)rr0r,r,r-┌ as_booleanWszFiniteDomain.as_booleanN)r1r2r3r4┌ is_Finiter5r@r?r/rHrIrJr,r,r,r-r6=s r6c@sXeZdZdZddДZeddДГZeddДГZedd ДГZed dДГZ dd ДZ ddДZdS)┌SingleFiniteDomainzi A FiniteDomain over a single symbol/set Example: The possibilities of a *single* die roll. cCs*t|tГst|tГst|О}tа|||бSr7)┌ isinstancerrr┌__new__)┌cls┌symbol┌setr,r,r-rNbs zSingleFiniteDomain.__new__cCs |jdSrArBr0r,r,r-rPhszSingleFiniteDomain.symbolcCs t|jГSr7йrrPr0r,r,r-r@lszSingleFiniteDomain.symbolscCs |jdSйNщrBr0r,r,r-rQpszSingleFiniteDomain.setcstЗfddДИjDГОS)Ncsg|]}tИj|ffГСqSr,й┌ frozensetrPйr9┌elemr0r,r-rEvr=z/SingleFiniteDomain.elements..)rrQr0r,r0r-r?tszSingleFiniteDomain.elementscsЗfddДИjDГS)Nc3s|]}tИj|ffГVqdSr7rUrWr0r,r-r<yr=z.SingleFiniteDomain.__iter__..йrQr0r,r0r-rIxszSingleFiniteDomain.__iter__cCs$t|Гd\}}||jko"||jvSrA)┌tuplerPrQ)r*rGr:r;r,r,r-rH{szSingleFiniteDomain.__contains__N)r1r2r3r4rNr5rPr@rQr?rIrHr,r,r,r-rL[s rLc@s$eZdZdZddДZeddДГZdS)┌ProductFiniteDomainzК A Finite domain consisting of several other FiniteDomains Example: The possibilities of the rolls of three independent dice cCst|jО}ddД|DГS)Ncss|]}t|ГVqdSr7)r$)r9┌itemsr,r,r-r<Йr=z/ProductFiniteDomain.__iter__..)r┌domains)r*┌proditerr,r,r-rIЗs zProductFiniteDomain.__iter__cCst|ОSr7rr0r,r,r-r?ЛszProductFiniteDomain.elementsN)r1r2r3r4rIr5r?r,r,r,r-r[Аsr[c@sDeZdZdZddДZddДZddДZdd ДZed dДГZ dd ДZ dS)┌ConditionalFiniteDomainzЭ A FiniteDomain that has been restricted by a condition Example: The possibilities of a die roll under the condition that the roll is even. cCs"|dur|St|Г}tа|||бS)zH Create a new instance of ConditionalFiniteDomain class T)r%rrN)rO┌domain┌ condition┌condr,r,r-rNШszConditionalFiniteDomain.__new__cCsB|jаt|Гб}|dvr|S|jr.|j|jkStdt|ГГВdS)zя Test the value. If value is boolean, return it. If value is equality relational (two objects are equal), return it with left-hand side being equal to right-hand side. Otherwise, raise ValueError exception. )TFzUndecidable if %sN)ra┌xreplacer/Zis_Equality┌lhs┌rhs┌ ValueError┌str)r*rXr;r,r,r-┌_testбszConditionalFiniteDomain._testcCs||jvo|а|бSr7)┌ fulldomainrhrFr,r,r-rHоsz$ConditionalFiniteDomain.__contains__csЗfddДИjDГS)Nc3s|]}Иа|бr|VqdSr7йrhrWr0r,r-r<▓r=z3ConditionalFiniteDomain.__iter__..)rir0r,r0r-rI▒sz ConditionalFiniteDomain.__iter__cs2tИjtГr&tЗfddДИjjDГОStdГВdS)Ncs&g|]}tИjj|ffГИvr|СqSr,)rVrirPrWr0r,r-rE╖s z/ConditionalFiniteDomain.set..z7Not implemented on multi-dimensional conditional domain)rMrirLrrQ┌NotImplementedErrorr0r,r0r-rQ┤s zConditionalFiniteDomain.setcCs tа|бSr7)r6rJr0r,r,r-rJ╜sz"ConditionalFiniteDomain.as_booleanN)r1r2r3r4rNrhrHrIr5rQrJr,r,r,r-r_Рs r_c@sРeZdZddДZeddДГZeeddДГГZddДZ ed d ДГZ eddДГZed dДГZeddДГZ eddДГZeddДГZddДZddДZdS)┌SingleFiniteDistributioncGs ttt|ГГ}tj|g|вRОSr7)┌list┌maprrrN)rOrCr,r,r-rN┬sz SingleFiniteDistribution.__new__cGsdSr7r,rBr,r,r-┌check╞szSingleFiniteDistribution.checkcs"ИjrtИГSЗfddДИjDГS)Ncsi|]}|Иа|бУqSr,й┌pmf)r9┌kr0r,r-┌ ╧r=z1SingleFiniteDistribution.dict..)┌is_symbolicr'rQr0r,r0r-r/╩szSingleFiniteDistribution.dictcGs tГВdSr7йrkйr*rCr,r,r-rq╤szSingleFiniteDistribution.pmfcCs tГВdSr7rur0r,r,r-rQ╘szSingleFiniteDistribution.setcCs|jjSr7)r/┌valuesr0r,r,r-┌╪r=z!SingleFiniteDistribution.cCs|jjSr7)r/r\r0r,r,r-rx┘r=cCsdS)NFr,r0r,r,r-rx┌r=cCs|jjSr7)r/rIr0r,r,r-rx█r=cCs|jjSr7)r/┌__getitem__r0r,r,r-rx▄r=cGs |j|ОSr7rprvr,r,r-r.▐sz!SingleFiniteDistribution.__call__cCs ||jvSr7rYrFr,r,r-rHсsz%SingleFiniteDistribution.__contains__N)r1r2r3rN┌staticmethodror5rr/rqrQrwr\rtrIryr.rHr,r,r,r-rl┴s rlc@sТeZdZdZdZddДZddДZddДZd d ДZe ddДГZ e d!ddДГZe ddДГZe ddДГZ d"ddДZddДZddДZddДZd#dd ДZdS)$┌FinitePSpacezd A Finite Probability Space Represents the probabilities of a finite number of events. TcCs2ddД|абDГ}t|Г}tа|||б}||_|S)NcSsi|]\}}t|Гt|ГУqSr,rйr9┌keyr;r,r,r-rsєs z(FinitePSpace.__new__..)r\rr rN┌_density)rOr`┌densityZpublic_density┌objr,r,r-rNЄs zFinitePSpace.__new__cCsLt|Г}|j}tt|абГdtГr2|а|tjбS|аt |ГddtjбS)NrrT) rr~rMrm┌keysr┌getr┌ZerorZ)r*rXrr,r,r-┌prob_of√s zFinitePSpace.prob_ofcs*tЗfddДt|ГDГГsJВtИj|ГS)Nc3s|]}|jИjvVqdSr7)rPr@)r9┌rr0r,r-r<r=z%FinitePSpace.where..)┌allr#r_r`йr*rar,r0r-┌whereszFinitePSpace.wherecCsPt||jГ}tГ}|jD]2}|аt|Гб}|а|б}|а|tj б|||<q|Sr7) r%rwr)r`rcr/rДrВrrГ)r*┌expr┌drXr;┌probr,r,r-┌compute_densitys zFinitePSpace.compute_densitycCsH|а|б}tj}g}t|ГD]"}||}||7}|а||fбqt|ГSr7)rМrrГ┌sorted┌appendr/)r*rЙrК┌cum_prob┌cdfr}rЛr,r,r-┌compute_cdfs zFinitePSpace.compute_cdfFcCs<|а|б}t|абГ}t|ddДdН}|r8ddД|DГ}|S)NcSs|dSrSr,)Zval_cumprobr,r,r-rxr=z)FinitePSpace.sorted_cdf..)r}cSsg|]\}}|t|ГfСqSr,)┌float)r9┌vrПr,r,r-rE!s z+FinitePSpace.sorted_cdf..)rСrmr\rН)r*rЙZpython_floatrРr\┌sorted_itemsr,r,r-┌ sorted_cdfs zFinitePSpace.sorted_cdfcs6|а|б}tdddНЙtИtЗfddД|абDГГГS)N┌tTй┌realc3s&|]\}}tt|ИГ|VqdSr7)rrйr9rrrУйrЦr,r-r<*r=z?FinitePSpace.compute_characteristic_function..йrМr r┌sumr\йr*rЙrКr,rЪr-┌compute_characteristic_function%s z,FinitePSpace.compute_characteristic_functioncs6|а|б}tdddНЙtИtЗfddД|абDГГГS)NrЦTrЧc3s"|]\}}t|ИГ|VqdSr7rrЩrЪr,r-r<1r=zBFinitePSpace.compute_moment_generating_function..rЫrЭr,rЪr-┌"compute_moment_generating_function,s z/FinitePSpace.compute_moment_generating_functionNcsЪ|pИj}tИ|ГЙЗfddДИjDГ}tИttfГr\ddДИjDГ}ЗfddДИjDГ}n$ЗfddДИjDГ}ddДИjDГ}tddДt|||ГDГГS)Ncsg|]}Иа|бСqSr,йrДrWr0r,r-rE6r=z4FinitePSpace.compute_expectation..cSsg|]}t|ГddСqS)rrT)rZrWr,r,r-rE8r=csg|]}Иаt|ГбСqSr,йrcr/rWйrЙr,r-rE9r=csg|]}Иаt|ГбСqSr,rбrWrвr,r-rE;r=cSsg|]}dСqSйTr,rWr,r,r-rE<r=cSs*g|]"\}}}t|||ftjdfГСqSrг)rrrГ)r9rЛrXZblvr,r,r-rE=s )rwr%r`rMrrrЬ┌zip)r*rЙ┌rvs┌kwargs┌probsZparse_domainZboolsr,)rЙr*r-┌compute_expectation3s z FinitePSpace.compute_expectationcCs^|а|б}tdddН}t|dk|dkBff}|абD]\}}||||kff}q4t|t|ОГS)N┌pTrЧrrT)rСr r r\rr)r*rЙrРrйrQr}┌valuer,r,r-┌compute_quantile@s zFinitePSpace.compute_quantilecsоtddДtИГDГГ}tИГ}|аИjбs@tdt|ИjГГВtИtГrО|j аИj j бsОtИjtГrlИj nИjЙtЗЗЗfddДИj DГГSttЗfddДИаИбDГГГS)Ncss|]}|jVqdSr7)rP)r9┌rsr,r,r-r<Ir=z+FinitePSpace.probability..z)Cannot compare foreign random symbols, %sc3s<|]4}tИа|бИаИt|ГddбftjdfГVqdS)rrTTN)rrД┌subsrmrrГrWйra┌rvr*r,r-r<Qs ■ ■c3s|]}Иа|бVqdSr7rаrWr0r,r-r<Tr=)rVr#r%┌issubsetr@rfrgrMrZfree_symbolsr`rerrdrЬrrИ)r*raZcond_symbolsrbr,rоr-┌probabilityHs ■zFinitePSpace.probabilitycs8|а|бЙ|а|бЙЗЗfddД|jабDГ}tИ|ГS)Ncs$i|]\}}Иа|бr||ИУqSr,rjr|йr`rЛr,r-rsYs z2FinitePSpace.conditional_space..)rИr▒r~r\r{йr*rarr,r▓r-┌conditional_spaceVs zFinitePSpace.conditional_spacer,┌scipycCs|j|jа|||бiS)zo Internal sample method Returns dictionary mapping RandomSymbol to realization value. )rк┌distribution┌sample)r*┌sizeZlibrary┌seedr,r,r-r╖]szFinitePSpace.sample)F)N)r,r╡N)r1r2r3r4rKrNrДrИrМrrСrХrЮrЯrиrлr▒r┤r╖r,r,r,r-r{ъs& r{c@sЦeZdZdZeddДГZeddДГZeddДГZdd ДZee d dДГГZ e dd ДГZe ddДГZddДZ ddДZddДZdddДZddДZddДZdS)┌SingleFinitePSpacea A single finite probability space Represents the probabilities of a set of random events that can be attributed to a single variable/symbol. This class is implemented by many of the standard FiniteRV types such as Die, Bernoulli, Coin, etc.... cCst|j|jjГSr7)rLrPr╢rQr0r,r,r-r`pszSingleFinitePSpace.domaincCs|jjS)zГ Helper property to check if the distribution of the random variable is having symbolic dimension. )r╢rtr0r,r,r-┌_is_symbolictszSingleFinitePSpace._is_symboliccCs |jdSrSrBr0r,r,r-r╢}szSingleFinitePSpace.distributioncCs|jа|бSr7)r╢rqйr*rЙr,r,r-rqБszSingleFinitePSpace.pmfcsЗfddДИjjабDГS)Ncs i|]\}}tИj|fГ|УqSr,rR)r9r;rЛr0r,r-rsЗs z/SingleFinitePSpace._density..)r╢r/r\r0r,r0r-r~Дs zSingleFinitePSpace._densitycCs~|jr^|а|б}tdddН}tdГ}t|t||Гtt||Г||jdj|jdj fГГSt ||jГ}t|j |jГа|бSйNrЦTrЧ┌kirT)r╗rМr rrrrrC┌low┌highr%rwr{r`r╢rЮйr*rЙrКrЦr╛r,r,r-rЮКs :z2SingleFinitePSpace.compute_characteristic_functioncCsz|jrZ|а|б}tdddН}tdГ}t|t||Гt||Г||jdj|jdjfГГSt ||j Г}t|j|j Га|бSr╜)r╗rМr rrrrCr┐r└r%rwr{r`r╢rЯr┴r,r,r-rЯФs 6z5SingleFinitePSpace.compute_moment_generating_functioncCs.|jrtdГВt||jГ}t|j|jГа|бS)NzГComputing quantile for random variables with symbolic dimension because the bounds of searching the required value is undetermined.)r╗rkr%rwr{r`r╢rлr╝r,r,r-rлЮsz#SingleFinitePSpace.compute_quantilec Csа|jrАtt|ГГd}tdddН}t|ttfГs4dn |а||б}t|t |а |бt||jdj k||jdjk|ГftjdfГГSt||jГ}t|j|jГа|бS)NrrrTй┌integerrT)r╗rmr#r rMrrrнrrrqrrCr┐r└rrГr%rwr{r`r╢rМ)r*rЙrпrrrbr,r,r-rМжs z"SingleFinitePSpace.compute_densitycCsb|jrB|а|б}tdГ}tdГ}t|t||Г||jdj|fГГSt||jГ}t |j |jГа|бS)Nrrr╛rT) r╗rМr rrrCr┐r%rwr{r`r╢rС)r*rЙrКrrr╛r,r,r-rС▓s "zSingleFinitePSpace.compute_cdfNcKs└|jrРt|Гd}tdddН}|а||б}t|ttfГs.)r╗rИr▒r~r\r{r│r,r▓r-r┤╥s z$SingleFinitePSpace.conditional_space)N)r1r2r3r4r5r`r╗r╢rqrr~rЮrЯrлrМrСrиr▒r┤r,r,r,r-r║fs* r║c@sLeZdZdZeddДГZeeddДГГZeeddДГГZdd ДZ d dДZ dS) ┌ProductFinitePSpacezG A collection of several independent finite probability spaces cCstddД|jDГОS)NcSsg|] }|jСqSr,)r`йr9┌spacer,r,r-rEшr=z.ProductFinitePSpace.domain..)r[┌spacesr0r,r,r-r`цszProductFinitePSpace.domaincCs`tddД|jDГО}i}|D]:}tt|ОГ\}}t|Г}t|О}|а|tjб|||<qt |ГS)NcSsg|]}t|jабГСqSr,)┌iterr~r\r╞r,r,r-rEэs z0ProductFinitePSpace._density..) rr╚rmrдr$rrВrrГr)r*r^rКr\┌elemsrзrXrЛr,r,r-r~ъs zProductFinitePSpace._densitycCs t|jГSr7)rr~r0r,r,r-rўszProductFinitePSpace.densitycCstа||бSr7)r{r▒rЗr,r,r-r▒№szProductFinitePSpace.probabilitycCstа||бSr7)r{rМr╝r,r,r-rМ sz#ProductFinitePSpace.compute_densityN)r1r2r3r4r5r`rr~rr▒rМr,r,r,r-r┼тs r┼N)@r4┌ itertoolsrZsympy.concrete.summationsrZsympy.core.basicrZsympy.core.cacherZsympy.core.functionrZsympy.core.mulrZsympy.core.numbersrr Zsympy.core.relationalrZsympy.core.singletonrZsympy.core.symbolr rZsympy.core.sympifyrZ&sympy.functions.elementary.exponentialrZ$sympy.functions.elementary.piecewiserZsympy.logic.boolalgrrZsympy.sets.setsrZsympy.core.containersrZsympy.core.logicrrrrZsympy.stats.rvrrrr r!r"r#r$r%r&r'r(r/r)r6rLr[r_rlr{r║r┼r,r,r,r-┌s< 8%1)||