diff options
author | Travis Oliphant <oliphant@enthought.com> | 2006-08-29 06:52:33 +0000 |
---|---|---|
committer | Travis Oliphant <oliphant@enthought.com> | 2006-08-29 06:52:33 +0000 |
commit | 936b14b2363d0b57aaa109b255d332d222704640 (patch) | |
tree | 3023a450c63efcc3d6441ba0945299859157efec /numpy/lib/shape_base.py | |
parent | 7b9df715cabe56f14d09a97b7b4b2bf524a3706e (diff) | |
download | numpy-936b14b2363d0b57aaa109b255d332d222704640.tar.gz |
Add axis argument to usages of take
Diffstat (limited to 'numpy/lib/shape_base.py')
-rw-r--r-- | numpy/lib/shape_base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index df0e9876d..925329cdf 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -25,7 +25,7 @@ def apply_along_axis(func1d,axis,arr,*args): indlist = range(nd) indlist.remove(axis) i[axis] = slice(None,None) - outshape = asarray(arr.shape).take(indlist) + outshape = asarray(arr.shape).take(indlist,0) i.put(ind, indlist) res = func1d(arr[tuple(i.tolist())],*args) # if res is a number, then we have a smaller output array |