diff options
| author | Travis Oliphant <oliphant@enthought.com> | 2006-08-05 07:50:50 +0000 |
|---|---|---|
| committer | Travis Oliphant <oliphant@enthought.com> | 2006-08-05 07:50:50 +0000 |
| commit | d372798ae6bf88c3a4414027341d2b0322941c68 (patch) | |
| tree | f0df1e8e8de70e6f0240cceee251838c31fcf864 /numpy/linalg/linalg.py | |
| parent | 3457ac34f84f502de194ad825c0db4cb0d56dc82 (diff) | |
| download | numpy-d372798ae6bf88c3a4414027341d2b0322941c68.tar.gz | |
Move location of flatnonzero
Diffstat (limited to 'numpy/linalg/linalg.py')
| -rw-r--r-- | numpy/linalg/linalg.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py index d80d36685..b2ab357b7 100644 --- a/numpy/linalg/linalg.py +++ b/numpy/linalg/linalg.py @@ -18,7 +18,7 @@ __all__ = ['solve', from numpy.core import array, asarray, zeros, empty, transpose, \ intc, single, double, csingle, cdouble, inexact, complexfloating, \ newaxis, ravel, all, Inf, dot, add, multiply, identity, sqrt, \ - maximum, nonzero, diagonal, arange, fastCopyAndTranspose, sum, \ + maximum, flatnonzero, diagonal, arange, fastCopyAndTranspose, sum, \ argsort from numpy.lib import triu from numpy.linalg import lapack_lite @@ -313,7 +313,7 @@ eigenvalue u[i]. Satisfies the equation dot(a, v[:,i]) = u[i]*v[:,i] else: w = wr+1j*wi v = array(vr, w.dtype) - ind = nonzero(wi != 0.0)[0] # indices of complex e-vals + ind = flatnonzero(wi != 0.0) # indices of complex e-vals for i in range(len(ind)/2): v[ind[2*i]] = vr[ind[2*i]] + 1j*vr[ind[2*i+1]] v[ind[2*i+1]] = vr[ind[2*i]] - 1j*vr[ind[2*i+1]] |