o tBh:?@sBdZddlZddlZddlZddlmZmZmZm Z dZ e dZ ddZ d d Zd=d d Zd dZd=ddZd>ddZd=ddZddZd>ddZefddZdefddZd=ddZefdd Zdefd!d"Zd#d$Zd%d&Zd'd(Zd>d)d*Zd?d+d,Zd-d.Z d/d0Z!d1d2Z"d3d4Z#d5d6Z$d=d7d8Z%d9d:Z&d;d<Z'dS)@zD Numerical/mathematical related functions. .. versionadded:: 2.1.0 N)UNSETiteratoriterator_with_default iteriteratee)addceilclampdividefloormax_max_bymeanmean_bymedianmin_min_by moving_meanmultiplypowerround_scaleslope std_deviationsum_sum_bysubtract transposevariancezscoreinfcCs||S)a Adds two numbers. Args: a (number): First number to add. b (number): Second number to add. Returns: number Example: >>> add(10, 5) 15 .. versionadded:: 2.1.0 .. versionchanged:: 3.3.0 Support adding two numbers when passed as positional arguments. .. versionchanged:: 4.0.0 Only support two argument addition. )abr!r!g/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/pydash/numerical.pyr/rcCt|S)a Sum each element in `collection`. Args: collection (list|dict|number): Collection to process or first number to add. Returns: number: Result of summation. Example: >>> sum_([1, 2, 3, 4]) 10 .. versionadded:: 2.1.0 .. versionchanged:: 3.3.0 Support adding two numbers when passed as positional arguments. .. versionchanged:: 4.0.0 Move iteratee support to :func:`sum_by`. Move two argument addition to :func:`add`. )r collectionr!r!r$rJr%rcCstddt||DS)a Sum each element in `collection`. If iteratee is passed, each element of `collection` is passed through a iteratee before the summation is computed. Args: collection (list|dict|number): Collection to process or first number to add. iteratee (mixed|number, optional): Iteratee applied per iteration or second number to add. Returns: number: Result of summation. Example: >>> sum_by([1, 2, 3, 4], lambda x: x ** 2) 30 .. versionadded:: 4.0.0 cs|]}|dVqdSrNr!).0resultr!r!r$ xzsum_by..)sumrr(iterateer!r!r$resrcCr&)a Calculate arithmetic mean of each element in `collection`. Args: collection (list|dict): Collection to process. Returns: float: Result of mean. Example: >>> mean([1, 2, 3, 4]) 2.5 .. versionadded:: 2.1.0 .. versionchanged:: 4.0.0 - Removed ``average`` and ``avg`` aliases. - Moved iteratee functionality to :func:`mean_by`. )rr'r!r!r$r{srcCst||t|S)a Calculate arithmetic mean of each element in `collection`. If iteratee is passed, each element of `collection` is passed through a iteratee before the mean is computed. Args: collection (list|dict): Collection to process. iteratee (mixed, optional): Iteratee applied per iteration. Returns: float: Result of mean. Example: >>> mean_by([1, 2, 3, 4], lambda x: x ** 2) 7.5 .. versionadded:: 4.0.0 )rlenr0r!r!r$rsrcCttj||S)a Round number up to precision. Args: x (number): Number to round up. precision (int, optional): Rounding precision. Defaults to ``0``. Returns: int: Number rounded up. Example: >>> ceil(3.275) == 4.0 True >>> ceil(3.215, 1) == 3.3 True >>> ceil(6.004, 2) == 6.01 True .. versionadded:: 3.3.0 )roundermathrx precisionr!r!r$rrcCs0|dur|}|}||kr|}|S||kr|}|S)a Clamps number within the inclusive lower and upper bounds. Args: x (number): Number to clamp. lower (number, optional): Lower bound. upper (number): Upper bound Returns: number Example: >>> clamp(-10, -5, 5) -5 >>> clamp(10, -5, 5) 5 >>> clamp(10, 5) 5 >>> clamp(-10, 5) -10 .. versionadded:: 4.0.0 Nr!)r7lowerupperr!r!r$r sr cCt||tjdS)a Divide two numbers. Args: dividend (int/float): The first number in a division. divisor (int/float): The second number in a division. Returns: int/float: Returns the quotient. Example: >>> divide(20, 5) 4.0 >>> divide(1.5, 3) 0.5 >>> divide(None, None) 1.0 >>> divide(5, None) 5.0 .. versionadded:: 4.0.0 r)call_math_operatoroperatortruediv)dividenddivisorr!r!r$r r cCr3)a Round number down to precision. Args: x (number): Number to round down. precision (int, optional): Rounding precision. Defaults to ``0``. Returns: int: Number rounded down. Example: >>> floor(3.75) == 3.0 True >>> floor(3.215, 1) == 3.2 True >>> floor(0.046, 2) == 0.04 True .. versionadded:: 3.3.0 )r4r5r r6r!r!r$r r9r cC t||dS)a Retrieves the maximum value of a `collection`. Args: collection (list|dict): Collection to iterate over. default (mixed, optional): Value to return if `collection` is empty. Returns: mixed: Maximum value. Example: >>> max_([1, 2, 3, 4]) 4 >>> max_([], default=-1) -1 .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Moved iteratee iteratee support to :func:`max_by`. default)r r(rEr!r!r$r  r cC*t|tr |}tt||t|dS)a Retrieves the maximum value of a `collection`. Args: collection (list|dict): Collection to iterate over. iteratee (mixed, optional): Iteratee applied per iteration. default (mixed, optional): Value to return if `collection` is empty. Returns: mixed: Maximum value. Example: >>> max_by([1.0, 1.5, 1.8], math.floor) 1.0 >>> max_by([{'a': 1}, {'a': 2}, {'a': 3}], 'a') {'a': 3} >>> max_by([], default=-1) -1 .. versionadded:: 4.0.0 key) isinstancedictvaluesmaxrpydr1r(r1rEr!r!r$r 6s r cCszt|}|dd}tddt||D}t|r%|t|d}|St|d}t|d}||||d}|S)a Calculate median of each element in `collection`. If iteratee is passed, each element of `collection` is passed through a iteratee before the median is computed. Args: collection (list|dict): Collection to process. iteratee (mixed, optional): Iteratee applied per iteration. Returns: float: Result of median. Example: >>> median([1, 2, 3, 4, 5]) 3 >>> median([1, 2, 3, 4]) 2.5 .. versionadded:: 2.1.0 rcsr)r*r!)r+retr!r!r$r-jr.zmedian..g?g?)r2sortedrrOis_oddint)r(r1lengthmiddler,leftrightr!r!r$rSs   rcCrC)a Retrieves the minimum value of a `collection`. Args: collection (list|dict): Collection to iterate over. default (mixed, optional): Value to return if `collection` is empty. Returns: mixed: Minimum value. Example: >>> min_([1, 2, 3, 4]) 1 >>> min_([], default=100) 100 .. versionadded:: 1.0.0 .. versionchanged:: 4.0.0 Moved iteratee iteratee support to :func:`min_by`. rD)rrFr!r!r$rvrGrcCrH)a Retrieves the minimum value of a `collection`. Args: collection (list|dict): Collection to iterate over. iteratee (mixed, optional): Iteratee applied per iteration. default (mixed, optional): Value to return if `collection` is empty. Returns: mixed: Minimum value. Example: >>> min_by([1.8, 1.5, 1.0], math.floor) 1.8 >>> min_by([{'a': 1}, {'a': 2}, {'a': 3}], 'a') {'a': 1} >>> min_by([], default=100) 100 .. versionadded:: 4.0.0 rI)rKrLrMminrrOr1rPr!r!r$rs rcCsVg}t|}t|dt|dD]}||||}t||kr(|t|q|S)aN Calculate moving mean of each element of `array`. Args: array (list): List to process. size (int): Window size. Returns: list: Result of moving average. Example: >>> moving_mean(range(10), 1) [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0] >>> moving_mean(range(10), 5) [2.0, 3.0, 4.0, 5.0, 6.0, 7.0] >>> moving_mean(range(10), 10) [4.5] .. versionadded:: 2.1.0 .. versionchanged:: 4.0.0 Rename to ``moving_mean`` and remove ``moving_average`` and ``moving_avg`` aliases. r)rUranger2appendr)arraysizer,iwindowr!r!r$rs rcCr<)a Multiply two numbers. Args: multiplier (int/float): The first number in a multiplication. multiplicand (int/float): The second number in a multiplication. Returns: int/float: Returns the product. Example: >>> multiply(4, 5) 20 >>> multiply(10, 4) 40 >>> multiply(None, 10) 10 >>> multiply(None, None) 1 .. versionadded:: 4.0.0 r)r=r>mul) multiplier multiplicandr!r!r$rrBrcs@t|r t|}|St|rfdd|D}|Sd}|S)a{ Calculate exponentiation of `x` raised to the `n` power. Args: x (number): Base number. n (number): Exponent. Returns: number: Result of calculation. Example: >>> power(5, 2) 25 >>> power(12.5, 3) 1953.125 .. versionadded:: 2.1.0 .. versionchanged:: 4.0.0 Removed alias ``pow_``. csg|]}t|qSr!)powr+itemnr!r$ szpower..N)rO is_numberrdis_list)r7rhr,r!rgr$rs   rcCs tt||S)a Round number to precision. Args: x (number): Number to round. precision (int, optional): Rounding precision. Defaults to ``0``. Returns: int: Rounded number. Example: >>> round_(3.275) == 3.0 True >>> round_(3.275, 1) == 3.3 True .. versionadded:: 2.1.0 .. versionchanged:: 4.0.0 Remove alias ``curve``. )r4roundr6r!r!r$r rGrcs"t|}||fdd|DS)a Scale list of value to a maximum number. Args: array (list): Numbers to scale. maximum (number): Maximum scale value. Returns: list: Scaled numbers. Example: >>> scale([1, 2, 3, 4]) [0.25, 0.5, 0.75, 1.0] >>> scale([1, 2, 3, 4], 1) [0.25, 0.5, 0.75, 1.0] >>> scale([1, 2, 3, 4], 4) [1.0, 2.0, 3.0, 4.0] >>> scale([1, 2, 3, 4], 2) [0.5, 1.0, 1.5, 2.0] .. versionadded:: 2.1.0 csg|]}|qSr!r!refactorr!r$riAzscale..)rN)r]maximum array_maxr!rmr$r'srcCsH|d|d}}|d|d}}||krt}|S||||}|S)aI Calculate the slope between two points. Args: point1 (list|tuple): X and Y coordinates of first point. point2 (list|tuple): X and Y cooredinates of second point. Returns: float: Calculated slope. Example: >>> slope((1, 2), (4, 8)) 2.0 .. versionadded:: 2.1.0 rr)INFINITY)point1point2x1y1x2y2r,r!r!r$rDsrcCstt|S)a\ Calculate standard deviation of list of numbers. Args: array (list): List to process. Returns: float: Calculated standard deviation. Example: >>> round(std_deviation([1, 18, 20, 4]), 2) == 8.35 True .. versionadded:: 2.1.0 .. versionchanged:: 4.0.0 Remove alias ``sigma``. )r5sqrtr)r]r!r!r$rasrcCr<)a Subtracts two numbers. Args: minuend (int/float): Value passed in by the user. subtrahend (int/float): Value passed in by the user. Returns: int/float: Result of the difference from the given values. Example: >>> subtract(10, 5) 5 >>> subtract(-10, 4) -14 >>> subtract(2, 0.5) 1.5 .. versionadded:: 4.0.0 r)r=r>sub)minuend subtrahendr!r!r$rxsrcCs>g}t|D]\}}t|D] \}}t|||g|}qq|S)a Transpose the elements of `array`. Args: array (list): List to process. Returns: list: Transposed list. Example: >>> transpose([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) [[1, 4, 7], [2, 5, 8], [3, 6, 9]] .. versionadded:: 2.1.0 )rrOset_)r]transyrowr7colr!r!r$rs rcs,t|fdd}t||S)a Calculate the variance of the elements in `array`. Args: array (list): List to process. Returns: float: Calculated variance. Example: >>> variance([1, 18, 20, 4]) 69.6875 .. versionadded:: 2.1.0 cst|dS)NrQ)r)r7avgr!r$varszvariance..var)rrO_map_value)r]rr!rr$rs rcs0t||}t|t|fdd|DS)a  Calculate the standard score assuming normal distribution. If iteratee is passed, each element of `collection` is passed through a iteratee before the standard score is computed. Args: collection (list|dict): Collection to process. iteratee (mixed, optional): Iteratee applied per iteration. Returns: float: Calculated standard score. Example: >>> results = zscore([1, 2, 3]) # [-1.224744871391589, 0.0, 1.224744871391589] .. versionadded:: 2.1.0 csg|]}|qSr!r!rersigr!r$riszzscore..)rOrrr)r(r1r]r!rr$rs rcCsn|s|}|s|}t|szt|}Wn tyYnwt|s2zt|}Wn ty1Ynw|||S)z.rounder_funccsg|]}|qSr!r!re)rr!r$rirozrounder..)rdrOrj is_iterable TypeError)rr7r8r,r!)rr8rr$r4s    r4r)r)r)(__doc__r5r>pydashrOhelpersrrrr__all__rrrrrrrrrr r r r r rrrrrrrrrrrrrrr=r4r!r!r!r$sD    %    #% !