ftjddlZddlZddlmZmZmZmZddlm cm Z ddl m Z ee dddd d d d ZdS) N)array_namespacexp_capabilities xp_device_length_nonmasked)_axis_nan_policy_factoryc|SN)xs T/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/scipy/stats/_variation.pyrsac|fSr r )r _s r rrsA4r) n_outputsresult_to_tuple propagateF)keepdimsct|}||}|||d}d}|t|||jt |}t jdd5tj 5tj d| ||}| ||}|||z z dz} || z|z } dddn #1swxYwYdddn #1swxYwYd } tj||k||f| | } | jdkr| d n| S) a Compute the coefficient of variation. The coefficient of variation is the standard deviation divided by the mean. This function is equivalent to:: np.std(x, axis=axis, ddof=ddof) / np.mean(x) The default for ``ddof`` is 0, but many definitions of the coefficient of variation use the square root of the unbiased sample variance for the sample standard deviation, which corresponds to ``ddof=1``. The function does not take the absolute value of the mean of the data, so the return value is negative if the mean is negative. Parameters ---------- a : array_like Input array. axis : int or None, optional Axis along which to calculate the coefficient of variation. Default is 0. If None, compute over the whole array `a`. nan_policy : {'propagate', 'raise', 'omit'}, optional Defines how to handle when input contains ``nan``. The following options are available: * 'propagate': return ``nan`` * 'raise': raise an exception * 'omit': perform the calculation with ``nan`` values omitted The default is 'propagate'. ddof : int, optional Gives the "Delta Degrees Of Freedom" used when computing the standard deviation. The divisor used in the calculation of the standard deviation is ``N - ddof``, where ``N`` is the number of elements. `ddof` must be less than ``N``; if it isn't, the result will be ``nan`` or ``inf``, depending on ``N`` and the values in the array. By default `ddof` is zero for backwards compatibility, but it is recommended to use ``ddof=1`` to ensure that the sample standard deviation is computed as the square root of the unbiased sample variance. Returns ------- variation : ndarray The calculated variation along the requested axis. Notes ----- There are several edge cases that are handled without generating a warning: * If both the mean and the standard deviation are zero, ``nan`` is returned. * If the mean is zero and the standard deviation is nonzero, ``inf`` is returned. * If the input has length zero (either because the array has zero length, or all the input values are ``nan`` and ``nan_policy`` is ``'omit'``), ``nan`` is returned. * If the input contains ``inf``, ``nan`` is returned. References ---------- .. [1] Zwillinger, D. and Kokoska, S. (2000). CRC Standard Probability and Statistics Tables and Formulae. Chapman & Hall: New York. 2000. Examples -------- >>> import numpy as np >>> from scipy.stats import variation >>> variation([1, 2, 3, 4, 5], ddof=1) 0.5270462766947299 Compute the variation along a given dimension of an array that contains a few ``nan`` values: >>> x = np.array([[ 10.0, np.nan, 11.0, 19.0, 23.0, 29.0, 98.0], ... [ 29.0, 30.0, 32.0, 33.0, 35.0, 56.0, 57.0], ... [np.nan, np.nan, 12.0, 13.0, 16.0, 16.0, 17.0]]) >>> variation(x, axis=1, ddof=1, nan_policy='omit') array([1.05109361, 0.31428986, 0.146483 ]) N)r)axisdtypedeviceignore)divideinvalidg?ct||}||j|jt |}||dk||||jS)Nrr)rasarrayinfrrwherecopysignnan)std_amean_amxp_xp_infs r special_casezvariation..special_caseys`eV,,++cgV\)FBSBS+TTyyCLL&$A$A37KKKr) fill_valuer )rr reshaperrrnperrstatewarningscatch_warnings simplefiltermeanstdxpx apply_wherendim) ar nan_policyddofrxpnr&r% correctionresultr)s r variationr=sr   B 1 A | JJq%  $QT222!')TU,, WWA +Xx 8 8 8--(:Q:S:S--h'''&&qt$$1t8ns* #f, ------------------------------LLL_dai5&/)f>>>F ))6"::v5s7D"AD8 DD D D DD"D)rrr)r.numpyr,scipy._lib._array_apirrrrscipy._lib.array_api_extra_libarray_api_extrar3_axis_nan_policyrr=r rr rDs )((((((((666666K1.?.?q6Uq6q6q6q6q6q6q6r