Commit 9e456bc7 authored by Raymond Hettinger's avatar Raymond Hettinger Committed by Miss Islington (bot)

bpo-36018: Add properties for mean and stdev (GH-12022)



Responding to suggestions on the tracker and some off-line suggestions.

Davin suggested that english named accessors instead of greek letters would result in more intelligible user code. Steven suggested that the parameters still need to be *mu* and *theta* which are used elsewhere (and I noted those parameter names are used in linked-to resources). 

Michael suggested proving-out the API by seeing whether it generalized to *Lognormal*.  I did so and found that Lognormal distribution parameters *mu* and *sigma*  do not represent the mean and standard deviation of the lognormal distribution (instead, they are for the underlying regular normal distribution).

Putting these ideas together, we have NormalDist parameterized by *mu* and *sigma* but offering English named properties for accessors.  That gives lets us match other API that access mu and sigma, it matches the external resources on the topic, gives us clear english names in user code. The API extends nicely to LogNormal where the parameters and the summary statistic accessors are not the same.


https://bugs.python.org/issue36018
parent a875ea58
......@@ -488,13 +488,17 @@ of applications in statistics, including simulations and hypothesis testing.
If *sigma* is negative, raises :exc:`StatisticsError`.
.. attribute:: mu
.. attribute:: mean
The mean of a normal distribution.
A read-only property representing the `arithmetic mean
<https://en.wikipedia.org/wiki/Arithmetic_mean>`_ of a normal
distribution.
.. attribute:: sigma
.. attribute:: stdev
The standard deviation of a normal distribution.
A read-only property representing the `standard deviation
<https://en.wikipedia.org/wiki/Standard_deviation>`_ of a normal
distribution.
.. attribute:: variance
......@@ -566,8 +570,8 @@ of applications in statistics, including simulations and hypothesis testing.
>>> birth_weights = NormalDist.from_samples([2.5, 3.1, 2.1, 2.4, 2.7, 3.5])
>>> drug_effects = NormalDist(0.4, 0.15)
>>> combined = birth_weights + drug_effects
>>> f'mu={combined.mu :.1f} sigma={combined.sigma :.1f}'
'mu=3.1 sigma=0.5'
>>> f'mean: {combined.mean :.1f} standard deviation: {combined.stdev :.1f}'
'mean: 3.1 standard deviation: 0.5'
.. versionadded:: 3.8
......
......@@ -740,6 +740,16 @@ class NormalDist:
raise StatisticsError('cdf() not defined when sigma is zero')
return 0.5 * (1.0 + erf((x - self.mu) / (self.sigma * sqrt(2.0))))
@property
def mean(self):
'Arithmetic mean of the normal distribution'
return self.mu
@property
def stdev(self):
'Standard deviation of the normal distribution'
return self.sigma
@property
def variance(self):
'Square of the standard deviation'
......
......@@ -2128,6 +2128,12 @@ class TestNormalDist(unittest.TestCase):
with self.assertRaises(statistics.StatisticsError):
Y.cdf(90)
def test_properties(self):
X = statistics.NormalDist(100, 15)
self.assertEqual(X.mean, 100)
self.assertEqual(X.stdev, 15)
self.assertEqual(X.variance, 225)
def test_unary_operations(self):
NormalDist = statistics.NormalDist
X = NormalDist(100, 12)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment