ó áp7]c @s:dZddlmZddlmZmZddlZd„Zdd„Z d„Z d„Z e d kr6dd l mZd „Zeed d gd dgeƒddƒeed d gddgeƒddƒejddddgddddgddddgddddgddddggƒZe ejd d dƒejd d dƒeƒZeeedƒejdgdgdgdggƒZe ejd d dƒejd d dƒeƒZeeedƒejddddggƒZe ejd d dƒejd d dƒeƒZeeedƒndS(s?Quantizing a continuous distribution in 2d Author: josef-pktd iÿÿÿÿ(tprint_function(trangetlmapNcCsV||Œ}||d|dƒ}||d|dƒ}||Œ}||||S(s‰helper function for probability of a rectangle in a bivariate distribution Parameters ---------- lower : array_like tuple of lower integration bounds upper : array_like tuple of upper integration bounds cdf : callable cdf(x,y), cumulative distribution function of bivariate distribution how does this generalize to more than 2 variates ? ii((tlowertuppertcdftprobuutprobultproblutprobll((sIlib/python2.7/site-packages/statsmodels/sandbox/distributions/quantize.pytprob_bv_rectangle s   c Cst|tjƒs¬ttj|ƒ}t|ƒ}g}tttj|ƒtj|ƒkƒr¿xIt |ƒD]8}dg|}t dƒ||<|j |||ƒqjWq¿n|j d}|}tt|ƒƒ||ƒ}|jƒ}x)t |ƒD]}tj|d|ƒ}qôW|S(sühelper function for probability of a rectangle grid in a multivariate distribution how does this generalize to more than 2 variates ? bins : tuple tuple of bin edges, currently it is assumed that they broadcast correctly itaxisN(t isinstancetnptndarrayRtasarraytlentalltndimtonesRtNonetslicetappendtshapetprinttcopytdiff( tbinsRR tn_dimtbins_tdtslt cdf_valuestprobs((sIlib/python2.7/site-packages/statsmodels/sandbox/distributions/quantize.pyt prob_mv_grids"  '    c s(tj|ƒ}tj|ƒ}t|ƒd}t|ƒd}tjtj||fƒ}||dd…df|ƒ‰‡fd†}x}td|dƒD]h}x_td|dƒD]J}||f} |d|df} t| | |ƒ||d|dfNt( R RRtnanRRRR tisnantanytAssertionError( tbinsxtbinsyRtnxtnyR!tcdf_functxindtyindRR((R sIlib/python2.7/site-packages/statsmodels/sandbox/distributions/quantize.pytprob_quantize_cdf?s  ,c Cs tj|ƒ}tj|ƒ}t|ƒd}t|ƒd}tjtj||fƒ}xtd|dƒD]x}xotd|dƒD]Z}||||f}||d||df} t| ||ƒ||d|dfs8   !    ((0 00