From 936b14b2363d0b57aaa109b255d332d222704640 Mon Sep 17 00:00:00 2001
From: Travis Oliphant <oliphant@enthought.com>
Date: Tue, 29 Aug 2006 06:52:33 +0000
Subject: Add axis argument to usages of take

---
 numpy/lib/arraysetops.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'numpy/lib/arraysetops.py')

diff --git a/numpy/lib/arraysetops.py b/numpy/lib/arraysetops.py
index 7bd666029..cc9067712 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)
+        aux = ar.take(perm,0)
         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)
-    aux2 = tt.take(perm)
+    aux = ar.take(perm,0)
+    aux2 = tt.take(perm,0)
     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 )
+    return flag.take( indx , 0)
 
 ##
 # 03.11.2005, c
-- 
cgit v1.2.1