diff options
Diffstat (limited to 'numpy/core/numeric.py')
| -rw-r--r-- | numpy/core/numeric.py | 10 |
1 files changed, 1 insertions, 9 deletions
diff --git a/numpy/core/numeric.py b/numpy/core/numeric.py index 5dbd1c232..6489be17a 100644 --- a/numpy/core/numeric.py +++ b/numpy/core/numeric.py @@ -134,15 +134,7 @@ def argwhere(a): is a sequence of indices into a. This sequence must be converted to a tuple in order to be used to index into a. """ - if a.ndim == 0: - return array([],dtype=intp) - else: - b = a.nonzero() - retarr = empty((b[0].shape[0],a.ndim),dtype=intp) - for k in xrange(a.ndim): - retarr[:,k] = b[k] - return retarr - + return transpose(a.nonzero()) _mode_from_name_dict = {'v': 0, 's' : 1, |