[Numpy-discussion] add axis to results of reduction (mean, min, ...)
Keith Goodman
kwgoodman at gmail.com
Thu Aug 6 12:22:55 EDT 2009
On Thu, Aug 6, 2009 at 9:18 AM, Robert Kern<robert.kern at gmail.com> wrote:
> On Thu, Aug 6, 2009 at 11:15, Keith Goodman<kwgoodman at gmail.com> wrote:
>> thanksOn Thu, Aug 6, 2009 at 9:07 AM, Robert Kern<robert.kern at gmail.com> wrote:
>>> On Thu, Aug 6, 2009 at 11:03, Keith Goodman<kwgoodman at gmail.com> wrote:
>
>>>>>> pylab.demean??
>>>> Type: function
>>>> Base Class: <type 'function'>
>>>> String Form: <function demean at 0x3c5c050>
>>>> Namespace: Interactive
>>>> File: /usr/lib/python2.6/dist-packages/matplotlib/mlab.py
>>>> Definition: pylab.demean(x, axis=0)
>>>> Source:
>>>> def demean(x, axis=0):
>>>> "Return x minus its mean along the specified axis"
>>>> x = np.asarray(x)
>>>> if axis:
>>>> ind = [slice(None)] * axis
>>>> ind.append(np.newaxis)
>>>> return x - x.mean(axis)[ind]
>>>> return x - x.mean(axis)
>>>
>>> Ouch! That doesn't handle axis=-1.
>>>
>>> if axis != 0:
>>> ind = [slice(None)] * x.ndim
>>> ind[axis] = np.newaxis
>>
>> Hey, didn't you warn us about the dangers of "if arr" the other day?
>
> Yes, but actually that wasn't quite the problem. "if axis:" would have
> been fine, if a bit obscure, as long as the body was fixed to not
> expect axis>0.
Oh, of course. Thanks for pointing that out. Now I can fix a bug in my own code.
More information about the NumPy-Discussion
mailing list