K iddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZmZmZdd lmZmZmZmZmZmZm Z dd l!m"Z"e d d d giZ#edZ$e$jKe de fdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKe de fdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefdZ&e$jKedefd Z&e$jKedefd!Z&e$jKedefd"Z&e$jKe de fd#Z&e$jKe"de"fd$Z&y%)&)singledispatch)Dummy)exp)lambdify) import_module)DiscreteDistributionHandmade)SingleContinuousDistribution) ChiSquaredDistributionExponentialDistributionGammaDistributionLogNormalDistributionNormalDistributionParetoDistributionUniformDistributionBetaDistributionStudentTDistributionCauchyDistribution)GeometricDistributionLogarithmicDistributionNegativeBinomialDistributionPoissonDistributionSkellamDistributionYuleSimonDistributionZetaDistribution)SingleFiniteDistributionscipyfromliststats) import_kwargscyNdistsizeseeds g/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sympy/stats/sampling/sample_scipy.pydo_sample_scipyr(s r$cPddl}td}t||j|ddgGfdd|jj }|t |jjt |jjd}|j||S) NrznumpyrceZdZfdZy)_..scipy_pdfc|Sr!r")r$x handmade_pdfs r'_pdfz_..scipy_pdf._pdf$ ? "r)N)__name__ __module__ __qualname__r2)r1sr' scipy_pdfr.# #r)r7abnamer% random_state) scipy.statsrrpdfr rv_continuousfloatset_inf_suprvs)r$r%r&rr+r7scipy_rvr1s @r'_rHs c AAtxx{Wg,>?L#EKK--#5/ /kCH <)rrchi2rFrBkr#s r'rHrH,s. ;;    5=t$  OOr)ctjjjdt |j z ||S)N)scaler%r>)rrexponrFrBrater#s r'rHrH2s5 ;;   q5+;';$UY ZZr)ctjjjt |j t |j ||S)N)r:rPr%r>)rrgammarFrBrMthetar#s r'rHrH8s; ;;   5=djj8IPTcg hhr)ctjjjt t |j t |j||S)N)rPsr%r>)rrlognormrFrBrmeanstdr#s r'rHrH>s? ;;   " "s499~)>%/X\ko " ppr)ctjjjt |j t |j ||SN)locrPr%r>)rrnormrFrBrYrZr#s r'rHrHDs9 ;;    E$))$4E$((ORVei  jjr)ctjjjt |j t |j ||S)N)r;rPr%r>)rrparetorFrBalphaxmr#s r'rHrHIs; ;;   ! !E$**$5U477^RVei ! jjr)ctjjjt |j ||SrJ)rrtrFrBnur#s r'rHrHOs* ;;==  dggT  MMr)ctjjjt |j t |j |j z ||Sr\)rruniformrFrBleftrightr#s r'rHrHTsH ;;   " "uTYY'7uTZZRVR[R[E[?\cgvz " {{r)ctjjjt |j t |j||S)N)r:r;r%r>)rrbetarFrBrar#s r'rHrHZs; ;;    % "3uTYY7Gdae  ffr)ctjjjt |j t |j ||Sr\)rrcauchyrFrBx0rTr#s r'rHrH`s9 ;;   ! !eDGGnE$**.scipy_pmfc|Sr!r")r$r0 handmade_pmfs r'_pmfz_..scipy_pmf._pmfor3r)N)r4r5r6rv)rusr' scipy_pmfrsnr8r)rwr9r=) r?rqrrr@rBrCrDrErF)r$r%r&rqr+rwrGrus @r'rHrHgsv' c AAtxx{Wg,>?L#K#5/53G)+H <)rrgeomrFrBrzr#s r'rHrHw, ;;    %-d  NNr)ctjjjt |j ||Sry)rrlogserrFrBrzr#s r'rHrH|s, ;;   ! !E$&&M4 ! PPr)ctjjjt |j t |j ||S)N)nrzr%r>)rrnbinomrFrBrrzr#s r'rHrHs7 ;;   ! !E$&&MU466]\` ! aar)ctjjjt |j ||S)N)mur%r>)rrpoissonrFrBlamdar#s r'rHrHs. ;;   " "eDJJ&7dQU " VVr)ctjjjt |j t |j ||S)N)mu1mu2r%r>)rrskellamrFrBrrr#s r'rHrHs8 ;;   " "uTXXE$((ORVei " jjr)ctjjjt |j ||S)N)rar%r>)rr yulesimonrFrBrhor#s r'rHrHs- ;; $ $5?TX $ YYr)ctjjjt |j ||S)N)r:r%r>)rrzipfrFrBrWr#s r'rHrHr|r)cddlm}|j}gg}}|jD]9\}}|j t ||j t |;|d||f} | j||S)NrrprG)r<valuesr=)r?rqdictitemsappendintrBrF) r$r%r&rqdensity_r0yrMvrGs r'rHrHsx(yyH rqA 1 Q q Aq6:H <rs$#6-(48---VVV4 gj'-CD 67 6 ( 68 6 01P "P2P 12[ #[3[ +,i i-i /0q !q1q ,-k k.k,-k k.k ./N N0N-.| |/| *+g g,g ,-m m.m 67 6 ( 68 6/0O !O1O12Q #Q3Q67b (b8b-.W W/W-.k k/k/0Z !Z1Z*+O O,O 23 6 $ 64 6r)