ó ¡Š.Yc@sPddlmZddlmZdZdd d„ƒYZdd d„ƒYZdS( iÿÿÿÿ(t defaultdict(tdeques0.2.1tDictIQMcBsDeZdZdddd„Zd„Zd„Zd„Zd„ZRS(smComputes the Interquartile Mean (IQM) of a dictionary containing containing number buckets (key) and the count of their occurrences (value). The number bucket supports variable precision: - round_digits is the digit supplied to the round() function. - tenth_precise=True will round to one less digit for the lowest 10%. For example: round_digits=-2, tenth_precise would result in numbers less than 100 being rounded to to the nearest 10th instead of 0 or 100. none_action determines how non-numbers are treated (all non-numbers are stored as None): - discard: discard them (default) - max: replace them with the maximum value - min: replace them with the minimum value This class is efficient for datasets in which many numbers are repeated. It should not be used for large datasets with a uniform distribution. cCsòt|_d|_|dkr'd}n|dkr<t}n=|rs|dkr]tdƒ‚nt|ƒd|_nt}|dkr‘t|_n|dkr¦d}n|d krÁtdƒ‚nt|jƒ|_ ||_ ||_ ||_ dS( Niÿÿÿÿis/tenth_precise only applies to round_digits < 0.i tdiscardtmaxtmins5none_action must be one of 'discard', 'max', or 'min'(sdiscardsmaxsmin( tFalsetas_intstNonetboundt ExceptiontabstTrueRt prep_datatdatat none_actiont round_digitst tenth_precise(tselfRRR((s-/usr/local/lib/python2.7/dist-packages/iqm.pyt__init__s*              cCs=|jr!ittƒd6dd6Sittƒd6dd6SdS(Nt num_countsitcount(RRtinttfloat(R((s-/usr/local/lib/python2.7/dist-packages/iqm.pyR =s cCsò|j}|j}|j}|j}|j}y|jƒ}WnnXykt|ƒ}|r¢|r„||kr„t||dƒ}nt||ƒ}t|ƒ}nt||ƒ}Wn d}nX||d|cd7<||dcd7eZdZdd„Zd„Zd„Zd„Zd„ZRS(sComputes the moving average of the Interquartile Mean (IQM) of a deque computed each period. none_action determines how non-numbers are treated (all non-numbers are stored as None): - discard: discard them (default) - max: replace them with the maximum value - min: replace them with the minimum value This class sacrifices accuracy for speed and low memory usage. cCs^|dkrd}n|dkr0tdƒ‚n||_t|ƒ|_t|jƒ|_dS(NRRRs5none_action must be one of 'discard', 'max', or 'min'(sdiscardsmaxsmin(RR RRtperiodRR R(RR0R((s-/usr/local/lib/python2.7/dist-packages/iqm.pyR°s    cCs,itd|jƒd6dd6dd6dd6S(Nttdeckit deck_countt iqm_countgtiqm_sum(RR0(R((s-/usr/local/lib/python2.7/dist-packages/iqm.pyR »scCsy|jƒ}WnnXyt|ƒ}Wn d}nX|j|}|dj|ƒ|dcd7<|d|jkr‹|j|ƒndS(s/For the key provided, add the num to its deque.R2R3iN(RRRRtappendR0R5(RRRR((s-/usr/local/lib/python2.7/dist-packages/iqm.pyR¿s  c Csy|j|}t|dƒ}|j}|ddkr:dSd|kræ|jdƒ}xd|krt|jdƒqXW|d krÉ|dkr«|d}||g|}qÙ|d}|g||}n|dc|8<|jƒn|ddkrtt|ƒ|dƒ}n8t d |dƒ} || | !}t|ƒt |ƒ}|d c|7<|d cd 7sš