diff options
| author | Huang, Guangtai <guangtai@amazon.com> | 2019-08-12 12:34:01 +0800 |
|---|---|---|
| committer | Huang, Guangtai <guangtai@amazon.com> | 2019-08-13 11:43:20 +0800 |
| commit | 3b3eaa60dc31677d1b19cb49e5c414d41ac4089a (patch) | |
| tree | 65df45e7f76fa9bdae959cae8343fea8deb8fbd8 /numpy/lib/arraysetops.py | |
| parent | cff02456fed4113924feef1d561e5e927aa2d86b (diff) | |
| download | numpy-3b3eaa60dc31677d1b19cb49e5c414d41ac4089a.tar.gz | |
fix unique
Diffstat (limited to 'numpy/lib/arraysetops.py')
| -rw-r--r-- | numpy/lib/arraysetops.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/lib/arraysetops.py b/numpy/lib/arraysetops.py index f3f4bc17e..c3e833f74 100644 --- a/numpy/lib/arraysetops.py +++ b/numpy/lib/arraysetops.py @@ -264,7 +264,7 @@ def unique(ar, return_index=False, return_inverse=False, # axis was specified and not None try: - ar = np.swapaxes(ar, axis, 0) + ar = np.moveaxis(ar, axis, 0) except np.AxisError: # this removes the "axis1" or "axis2" prefix from the error message raise np.AxisError(axis, ar.ndim) @@ -285,7 +285,7 @@ def unique(ar, return_index=False, return_inverse=False, def reshape_uniq(uniq): uniq = uniq.view(orig_dtype) uniq = uniq.reshape(-1, *orig_shape[1:]) - uniq = np.swapaxes(uniq, 0, axis) + uniq = np.moveaxis(uniq, 0, axis) return uniq output = _unique1d(consolidated, return_index, |