Commit f7a7f4ea authored by Jason Swails's avatar Jason Swails
Browse files

Fix the mean() and std() methods on container Quantities when those containers

are NOT numpy arrays.
parent b9a19a0c
......@@ -506,7 +506,7 @@ class Quantity(object):
except AttributeError:
if args or kwargs:
raise TypeError('Unsupported arguments for Quantity.mean')
mean = self.sum() / len(self._value)
mean = (self.sum() / len(self._value))._value
return Quantity(mean, self.unit)
def std(self, *args, **kwargs):
......@@ -526,7 +526,8 @@ class Quantity(object):
except AttributeError:
if args or kwargs:
raise TypeError('Unsupported arguments for Quantity.std')
mean = self.mean()
mean = self.mean()._value
var = 0
for val in self._value:
res = mean - val
var += res * res
......
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