B Y@shddlZddlmZmZdddddgZdddZdd dZd d Zd d Zej dfddZ dddZ dS)N) partition argpartitionrankdata nanrankdatarrpushcCs tt||S)z5Slow rankdata function used for unaccelerated dtypes.)_rankscipy_rankdata)aaxisr =lib/python3.7/site-packages/bottleneck/slow/nonreduce_axis.pyrscCs tt||S)z8Slow nanrankdata function used for unaccelerated dtypes.)r_nanrankdata_1d)r r r r r r scCshtj|dd}|dkr"|}d}|jdkr>|jtjdd}n&t|||}|jtjkrd|tj}|S)NF)copyrT)nparrayravelsizeZastypefloat64Zapply_along_axisdtype)Zfunc1dr r yr r r rs   rcCs>tj|jtjd}|tjt|}t||||<|S)N)r)remptyshaperfillnanisnanr)r ridxr r r r s   r c Cs||dkrtdt|}|j}|dks4||dkrBt|||}|dkr\|dddf}n |dkrh|St|}t|jdd}t|jdd}|tj |tj tj ddvxnt |jdD]\}|||k} tj || <|d|f} || || |f<|d|f} ||| <|| |f|| <qWWdQRX|dksT||dkrft||d|}|dkrx|dS|S) z(Slow push used for unaccelerated dtypes.Nz`axis` cannot be Nonerrignore)Zinvalid.) ValueErrorrrndimZrollaxisrrrrrZerrstaterange) r nr rr ZfidxZrecentcountirr r r r's:        averagecCs |dkrtd|tt|}|dkr2dnd}tj||d}tj|jtjd}tj |jtjd||<|dkr||dS||}tj d |dd |d d kf}| |}|d kr|Stj t |d t |f}|dkr||S|dkr||ddSd||||ddS)a rankdata(a, method='average') Assign ranks to data, dealing with ties appropriately. Ranks begin at 1. The `method` argument controls how ranks are assigned to equal values. See [1]_ for further discussion of ranking methods. Parameters ---------- a : array_like The array of values to be ranked. The array is first flattened. method : str, optional The method used to assign ranks to tied elements. The options are 'average', 'min', 'max', 'dense' and 'ordinal'. 'average': The average of the ranks that would have been assigned to all the tied values is assigned to each value. 'min': The minimum of the ranks that would have been assigned to all the tied values is assigned to each value. (This is also referred to as "competition" ranking.) 'max': The maximum of the ranks that would have been assigned to all the tied values is assigned to each value. 'dense': Like 'min', but the rank of the next highest element is assigned the rank immediately after those assigned to the tied elements. 'ordinal': All values are given a distinct rank, corresponding to the order that the values occur in `a`. The default is 'average'. Returns ------- ranks : ndarray An array of length equal to the size of `a`, containing rank scores. References ---------- .. [1] "Ranking", http://en.wikipedia.org/wiki/Ranking Examples -------- >>> from scipy.stats import rankdata >>> rankdata([0, 2, 3, 2]) array([ 1. , 2.5, 4. , 2.5]) >>> rankdata([0, 2, 3, 2], method='min') array([ 1, 2, 4, 2]) >>> rankdata([0, 2, 3, 2], method='max') array([ 1, 3, 4, 3]) >>> rankdata([0, 2, 3, 2], method='dense') array([ 1, 2, 3, 2]) >>> rankdata([0, 2, 3, 2], method='ordinal') array([ 1, 2, 4, 3]) )r%minmaxdenseordinalzunknown method "{0}"r)Z mergesortZ quicksort)Zkind)rrTNrr(rr'r&g?)rformatrrZasarrayZargsortrrZintpZarangeZr_ZcumsumZnonzerolen)r methodZalgoZsorterinvZobsr(r#r r r rSs(4"  r)N)N)r%) Znumpyrrr__all__rrrr infrrr r r r s  ,