diff options
| -rw-r--r-- | numpy/lib/arraysetops.py | 8 | ||||
| -rw-r--r-- | numpy/lib/shape_base.py | 2 |
2 files changed, 5 insertions, 5 deletions
diff --git a/numpy/lib/arraysetops.py b/numpy/lib/arraysetops.py index cc9067712..7bd666029 100644 --- a/numpy/lib/arraysetops.py +++ b/numpy/lib/arraysetops.py @@ -66,7 +66,7 @@ def unique1d( ar1, retindx = False ): if retindx: ar = numpy.array(ar1).ravel() perm = ar.argsort() - aux = ar.take(perm,0) + aux = ar.take(perm) flag = ediff1d( aux, 1 ) != 0 return perm.compress(flag), aux.compress(flag) else: @@ -113,8 +113,8 @@ def setmember1d( ar1, ar2 ): tt = concat( (zlike( ar1 ), zlike( ar2 ) + 1) ) perm = ar.argsort() - aux = ar.take(perm,0) - aux2 = tt.take(perm,0) + aux = ar.take(perm) + aux2 = tt.take(perm) flag = ediff1d( aux, 1 ) == 0 ii = numpy.where( flag * aux2 )[0] @@ -124,7 +124,7 @@ def setmember1d( ar1, ar2 ): indx = perm.argsort()[:len( ar1 )] - return flag.take( indx , 0) + return flag.take( indx ) ## # 03.11.2005, c diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index 925329cdf..df0e9876d 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,0) + outshape = asarray(arr.shape).take(indlist) i.put(ind, indlist) res = func1d(arr[tuple(i.tolist())],*args) # if res is a number, then we have a smaller output array |