O=d:?dZddlZddlZddlZddlmZmZmZm Z dZ e dZ dZ dZd"d Zd Zd"d Zd#d Zd"d ZdZd#dZefdZdefdZd"dZefdZdefdZdZdZdZd#dZd$dZdZ dZ!dZ"dZ#dZ$d"dZ%d Z&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 transposevariancezscoreinfc ||zS)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. )abs :D:\Sites\api_v1\venv\Lib\site-packages\pydash/numerical.pyrr/s 0 q5Lc t|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 collections r&rrJs0 *  r'cPtdt||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 c3&K|] }|dV dSrNr#).0results r& zsum_by..xs&JJVvayJJJJJJr')sumrr*iteratees r&rres+& JJ|J'I'IJJJ J JJr'c t|S)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)s r&rr{s, :  r'cBt||t|z 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 )rlenr2s r&rrs& *h ' '#j// 99r'c8ttj||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 )roundermathr x precisions r&r r s, 49a + ++r'c4||}|}||kr|}n||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 r#)r;loweruppers r&r r s62 }5yy  U  Hr'c:t||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)dividenddivisors r&r r s0 h1A1 E EEr'c8ttj||S)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 )r8r9r r:s r&r r s, 4:q) , ,,r'c$t||S)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)rr*rIs r&r r . *g . . ..r'ct|tr|}tt ||t j|S)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) isinstancedictvaluesmaxrpydr3r*r3rIs r&rr6sO.*d##)&&(( $Z99s|H?U?U V V VVr'cLt|}|dzdz }tdt||D}tj|r|t |dz }n8t |dz }t |dz }||||zdz }|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 rc3&K|] }|dV dSr-r#)r.rets r&r0zmedian..js&MM3AMMMMMMr'g?g?)r6sortedrrSis_oddint)r*r3lengthmiddler/leftrights r&rrSs*__FqjA FMM,z8*L*LMMMMMJ z&<C OO,6C<  FSL!!T"Z%66!; Mr'c$t||S)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`. rH)rrJs r&rrvrKr'ct|tr|}tt ||t j|S)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 rM)rOrPrQminrrSr3rTs r&rrsO.*d##)&&(( $Z99s|H?U?U V V VVr'cg}t|}t|dz t|dzD]D}|||z |}t||kr"|t |E|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)r[ranger6appendr)arraysizer/iwindows r&rrs|2F t99D 4!8SZZ!^ , ,((q4x!|$ v;;$   MM$v,, ' ' ' Mr'c:t||tjdS)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)rArBmul) multiplier multiplicands r&rrs0 j, a H HHr'ctj|rt|}n%tj|rfd|D}nd}|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_``. c0g|]}t|Sr#)pow)r.itemns r& zpower..s!---4#dA,,---r'N)rS is_numberrpis_list)r;rrr/s ` r&rrs[. }QQ Q----1--- Mr'c.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``. )r8roundr:s r&rr s. 5!Y ' ''r'cHt|}||z fd|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 cg|]}|zSr#r#)r.rqfactors r&rszscale..As , , ,dD6M , , ,r')rR)rfmaximum array_maxrzs @r&rr's40E I y F , , , ,e , , ,,r'cx|d|d}}|d|d}}||krt}n ||z ||z z }|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/s r&rrDsM$AYq B AYq B Rxxr'b2g& Mr'cDtjt|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``. )r9sqrtr)rfs r&rras( 9Xe__ % %%r'c:t||tjdS)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)rArBsub)minuend subtrahends r&rrxs, gz8< C CCr'cg}t|D]2\}}t|D]\}}tj|||g|}3|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 )rrSset_)rftransyrowr;cols r&rrs`" E5//113smm 1 1FAsHUQFC00EE 1 Lr'ct|fd}tj||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 c*t|z dS)NrV)r)r;avgs r&varzvariance..varsQWa   r')rrS_map_value)rfrrs @r&rrs_" u++C!!!!! 5<<  S ! ! & & ( ( . . 0 00r'ctj||}t|t|fd|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 c g|] }|z z  Sr#r#)r.rqrsigs r&rszzscore..s" 1 1 14TCZ3  1 1 1r')rSrrr)r*r3rfrrs @@r&r r sL( HZ * *E u++C   C 1 1 1 1 15 1 1 11r'c|s|}|s|}tj|s! t|}n#t$rYnwxYwtj|s! t|}n#t$rYnwxYw|||S)z.rounder_funcstD9$%% 11r'c&g|] }|Sr#r#)r.rqrs r&rszrounder..s#777Tll4((777r')rprSrt is_iterable TypeError)rr;r<r/rs` ` @r&r8r8sB ""I222222F }Qa    7777Q777FF    D  MsA A-,A-r)r)r)(__doc__r9rBpydashrShelpersrrrr__all__rr~rrrrrr r r r r rrrrrrrrrrrrrrr rAr8r#r'r&rs{  IIIIIIIIIIII : 5<<66KKKK,2::::,,,,,2" " " " JFFF6----2#////4!%eWWWW:    F#////4!%eWWWW8"""JIII6B((((4----::&&&.DDD2411122222@.r'