From 936b14b2363d0b57aaa109b255d332d222704640 Mon Sep 17 00:00:00 2001 From: Travis Oliphant Date: Tue, 29 Aug 2006 06:52:33 +0000 Subject: Add axis argument to usages of take --- numpy/lib/shape_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/lib/shape_base.py') 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 -- cgit v1.2.1