| Server IP : 93.86.61.54 / Your IP : 216.73.216.60 Web Server : Apache/2.4.62 (Ubuntu) System : Linux rasin.ddns.net 6.8.0-124-generic #124~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Tue May 26 21:05:19 UTC x86_64 User : www-data ( 33) PHP Version : 8.4.22 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /usr/lib/python3/dist-packages/sympy/stats/__pycache__/ |
Upload File : |
o
�8Va�, � @ sr d dl mZmZmZmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZmZmZm Z m Z m Z m Z m Z m!Z! d dl"m#Z# d dl$m%Z%m&Z& d dl'm(Z( d dl)m*Z* d d l+m,Z, d d
l-m.Z. G d d � d e!�Z/G d
d� de/e�Z0G dd� de �Z1G dd� de1e�Z2G dd� de1e �Z3G dd� de �Z4G dd� de e1�Z5G dd� de4e�Z6dS ) � )�Basic�sympify�symbols�Dummy�Lambda� summation� Piecewise�S�cacheit�Sum�exp�I�Ne�Eq�poly�series� factorial�And�floor)�PolynomialError)�!reduce_rational_inequalities_wrap) �NamedArgsMixin� SinglePSpace� SingleDomain�random_symbols�PSpace�ConditionalDomain� RandomDomain�
ProductDomain� Distribution)� Probability)�Range� FiniteSet)�Union)�Contains)�
filldedent)�_sympifyc @ � e Zd Zdd� ZdS )�DiscreteDistributionc G �
| j |� S �N��pdf��self�args� r0 �1/usr/lib/python3/dist-packages/sympy/stats/drv.py�__call__ �
z DiscreteDistribution.__call__N)�__name__�
__module__� __qualname__r2 r0 r0 r0 r1 r( � r( c @ s� e Zd ZdZejZdd� Zedd� �Z e
dd� �Z dd � Z d
d � Z
e
d d
� �Zdd� Zdd� Ze
dd� �Zdd� Zdd� Ze
dd� �Zdd� Zd d � Zd$d d �Zd!d"� Zd#S )%�SingleDiscreteDistributionz� Discrete distribution of a single variable.
Serves as superclass for PoissonDistribution etc....
Provides methods for pdf, cdf, and sampling
See Also:
sympy.stats.crv_types.*
c G s t tt|��}tj| g|�R � S r* )�list�mapr r �__new__)�clsr/ r0 r0 r1 r; $ s z"SingleDiscreteDistribution.__new__c G � d S r* r0 )r/ r0 r0 r1 �check( s z SingleDiscreteDistribution.checkc K sf t ddtd�}t ddtd�}| jj}| �|�}t|||t|�ffi |��}t|||kfd�}t||�S )zB Compute the CDF from the PDF.
Returns a Lambda.
�xT)�integerr< �z��realr< )r T) r r �set�infr, r r r r )r. �kwargsr? rA �
left_boundr, �cdfr0 r0 r1 � compute_cdf, s
z&SingleDiscreteDistribution.compute_cdfc C r= r* r0 �r. r? r0 r0 r1 �_cdf= � z SingleDiscreteDistribution._cdfc K �. |s
| � |�}|dur
|S | jdi |��|�S �z Cumulative density function Nr0 )rK rI )r. r? rF rH r0 r0 r1 rH @ �
z SingleDiscreteDistribution.cdfc K sN t ddtd�\}}| �|�}ttt| | �| || jj| jjf�}t ||�S )zV Compute the characteristic function from the PDF.
Returns a Lambda.
zx, tTrB )
r r r, r r r
rD rE �supr )r. rF r? �tr, �cfr0 r0 r1 � compute_characteristic_functionH s
(
z:SingleDiscreteDistribution.compute_characteristic_functionc C r= r* r0 �r. rQ r0 r0 r1 �_characteristic_functionS rL z3SingleDiscreteDistribution._characteristic_functionc K rM )z Characteristic function Nr0 )rU rS )r. rQ rF rR r0 r0 r1 �characteristic_functionV rO z2SingleDiscreteDistribution.characteristic_functionc K sP t ddd�}t ddd�}| �|�}tt|| �| || jj| jjf�}t||�S )NrQ T�rC r? �r@ )r r, r r rD rE rP r )r. rF rQ r? r, �mgfr0 r0 r1 �"compute_moment_generating_function^ s
$
z=SingleDiscreteDistribution.compute_moment_generating_functionc C r= r* r0 rT r0 r0 r1 �_moment_generating_functionf rL z6SingleDiscreteDistribution._moment_generating_functionc K s. |s
| � |�}|d ur
|S | jdi |��|�S )Nr0 )r[ rZ )r. rQ rF rY r0 r0 r1 �moment_generating_functioni s
z5SingleDiscreteDistribution.moment_generating_functionc K s^ t ddd�}t ddd�}| jj}| �|�}t||||ffi |��}|||kff}t|t|� �S )zG Compute the Quantile from the PDF.
Returns a Lambda.
r? TrX �prW )r rD rE r, r r r )r. rF r? r] rG r, rH rD r0 r0 r1 �compute_quantilep s
z+SingleDiscreteDistribution.compute_quantilec C r= r* r0 rJ r0 r0 r1 � _quantile~ rL z$SingleDiscreteDistribution._quantilec K rM rN )r_ r^ )r. r? rF �quantiler0 r0 r1 r` � rO z#SingleDiscreteDistribution.quantileTc K s� |rezDt ||�}tddd�}| �|�}|�� }t t||d|d ��� |�} d}
t|d �D ]} |
|�|| �| �|| � t| � 7 }
q-|
W S t yd t
|| � |� || j j
| j jffi |�� Y S w t|| � |� || j j
| j jffi |��S )z- Expectation of expression over distribution rQ TrW r � )r r r\ �degreer �removeO�range�coeff_monomialr r r r, rD rE rP r ) r. �expr�var�evaluaterF r] rQ rY �deg�taylor�result�kr0 r0 r1 � expectation� s.
* �
����z&SingleDiscreteDistribution.expectationc G r) r* r+ r- r0 r0 r1 r2 � r3 z#SingleDiscreteDistribution.__call__N)T)r4 r5 r6 �__doc__r �IntegersrD r; � staticmethodr> r
rI rK rH rS rU rV rZ r[ r\ r^ r_ r` rm r2 r0 r0 r0 r1 r8 s0
r8 c @ s e Zd ZdZdZdS )�DiscreteDomainze
A domain with discrete support with step size one.
Represented using symbols and Range.
TN)r4 r5 r6 rn � is_Discreter0 r0 r0 r1 rq � s rq c @ r' )�SingleDiscreteDomainc C � t | j| j�S r* )r$ �symbolrD �r. r0 r0 r1 �
as_boolean� s z SingleDiscreteDomain.as_booleanN�r4 r5 r6 rw r0 r0 r0 r1 rs � r7 rs c @ s e Zd ZdZedd� �ZdS )�ConditionalDiscreteDomainzb
Domain with discrete support of step size one, that is restricted by
some condition.
c C sB | j }t| j �dkrttd���t|�d }t| j|��| jj �S )Nra zJ
Multivariate conditional domains are not yet implemented.r )
r �len�NotImplementedErrorr% r9 r � condition� intersect�
fulldomainrD )r. �rvr0 r0 r1 rD � s �
�z ConditionalDiscreteDomain.setN)r4 r5 r6 rn �propertyrD r0 r0 r0 r1 ry � s ry c @ s@ e Zd ZdZdZedd� �Zdd� Zdd� Zdd � Z d
d � Z
d S )
�DiscretePSpaceTc C s | j | j� S r* )�densityr rv r0 r0 r1 r, � s zDiscretePSpace.pdfc sf t |�}t� fdd�|D ��sJ �t|�dkr ttd���t||d �}|�� jj�}t |d j
|�S )Nc 3 s � | ] }|j � jv V qd S r* )ru r ��.0�rrv r0 r1 � <genexpr>� s � z'DiscretePSpace.where.<locals>.<genexpr>ra zIMultivariate discrete
random variables are not yet supported.r ) r �allrz r{ r% r r} �domainrD rs ru )r. r| �rvs�conditional_domainr0 rv r1 �where� s �zDiscretePSpace.wherec C s t |t�}|rt|jd |jd �}z(| �|�j}|dks"|tju r&tjW S |dks0|| j jkr4tj
W S | � |�}W n= t yw ddl
m} |j|j }||�}t |t�s`ddlm} ||�}tddd�} t| |�}
|
�|�|
jd��}Y nw |d u r�t|�}|s�|S tj
| S ) Nr ra FT)r� )� DiscreteDistributionHandmaderA rW )�
isinstancer r r/ r� rD r �EmptySet�Zeror� �One� eval_probr{ �sympy.stats.rvr� �lhs�rhsr( �sympy.stats.drv_typesr� r �SingleDiscretePSpace� probability� __class__�valuer ) r. r| �
complement�_domain�probr� rf �densr� rA �spacer0 r0 r1 r� � s0
�
zDiscretePSpace.probabilityc s� t �j�d }t|t�r8tddd�}dd� |jD �\}}}�j�||| �}t|||| || d f��� }|S t|t �rPt
|�j�� t � fdd�|D ��}|S t|t �rct �fd d�|jD ��}|S d S )
Nr �nTrX c s s � | ]}|V qd S r* r0 r� r0 r0 r1 r� � s � z+DiscretePSpace.eval_prob.<locals>.<genexpr>ra c 3 s � | ]}� |�V qd S r* r0 �r� r? r+ r0 r1 r� � s � c 3 s � | ]}� � |�V qd S r* )r� r� rv r0 r1 r� � s � )
r9 r r� r! r/ r, �replacer �doitr" r �sumr# ) r. r� �symr� rE rP �step�summandr r0 )r, r. r1 r� � s*
���
�zDiscretePSpace.eval_probc C sH t t| j�| j| �|� �}|�dd� | jD ��}t| j|�}t ||�S )Nc S � i | ]}||j �qS r0 �ru �r� r r0 r0 r1 �
<dictcomp> � z4DiscretePSpace.conditional_space.<locals>.<dictcomp>)
r �tupler r, r� �xreplace�valuesry r� r� )r. r| r� r� r0 r0 r1 �conditional_space s
z DiscretePSpace.conditional_spaceN) r4 r5 r6 �is_realrr r� r, r� r� r� r� r0 r0 r0 r1 r� � s
r� c @ r' )�ProductDiscreteDomainc C s t dd� | jD �� S )Nc S s g | ]}|j �qS r0 )rw )r� r� r0 r0 r1 �
<listcomp> s z4ProductDiscreteDomain.as_boolean.<locals>.<listcomp>)r �domainsrv r0 r0 r1 rw s z ProductDiscreteDomain.as_booleanNrx r0 r0 r0 r1 r�
r7 r� c @ sh e Zd ZdZdZedd� �Zedd� �Zdd
d �Zdd d
�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd S )r� z> Discrete probability space over a single univariate variable Tc C s | j jS r* )� distributionrD rv r0 r0 r1 rD s zSingleDiscretePSpace.setc C rt r* )rs ru rD rv r0 r0 r1 r� s zSingleDiscretePSpace.domainr0 �scipyNc C s | j | jj|||d�iS )zp
Internal sample method.
Returns dictionary mapping RandomSymbol to realization value.
)�library�seed)r� r� �sample)r. �sizer� r� r0 r0 r1 r� s zSingleDiscretePSpace.samplec K s� |p| j f}| j |vr
|S t|�}|�dd� |D ��}| j j}z| jj||fd|i|��W S tyI t|| j || j j
| j j ffi |�� Y S w )Nc S r� r0 r� r� r0 r0 r1 r� ( r� z<SingleDiscretePSpace.compute_expectation.<locals>.<dictcomp>rh ) r� r& r� ru r� rm r{ r r, rD rE rP )r. rf r� rh rF r? r0 r0 r1 �compute_expectation" s
�
��z(SingleDiscretePSpace.compute_expectationc K �6 || j krtddd�}t|| jj|fi |���S t� �)Nr? TrW )r� r r r� rH r{ )r. rf rF r? r0 r0 r1 rI 2 �
z SingleDiscretePSpace.compute_cdfc K s || j kr| jS t� �r* )r� r� r{ )r. rf rF r0 r0 r1 �compute_density9 s
z$SingleDiscretePSpace.compute_densityc K r� �NrQ TrW )r� r r r� rV r{ �r. rf rF rQ r0 r0 r1 rS > r� z4SingleDiscretePSpace.compute_characteristic_functionc K r� r� )r� r r r� r\ r{ r� r0 r0 r1 rZ E r� z7SingleDiscretePSpace.compute_moment_generating_functionc K r� )Nr] TrW )r� r r r� r` r{ )r. rf rF r] r0 r0 r1 r^ L r� z%SingleDiscretePSpace.compute_quantile)r0 r� N)NT)r4 r5 r6 rn r� r� rD r� r� r� rI r� rS rZ r^ r0 r0 r0 r1 r� s
r� N)7�sympyr r r r r r r r r
r r r
r r r r r r r �sympy.polys.polyerrorsr �sympy.stats.crvr r� r r r r r r r r r � sympy.stats.symbolic_probabilityr �sympy.sets.fancysetsr! r" �sympy.sets.setsr# �sympy.sets.containsr$ �sympy.utilitiesr% �sympy.core.sympifyr& r( r8 rq rs ry r� r� r� r0 r0 r0 r1 �<module> s& T , F