[Numpy-discussion] sort bug

Anton Sherwood bronto at pobox.com
Sat Apr 28 23:47:36 EDT 2007


Travis Oliphant wrote:
> One approach is to use argsort to create an index list of sorted 
> eigenvalues and then sort the eig and eigvector arrays before zipping 
> them together in a list of tuples.
> 
> eig, val = numpy.linalg.eig(a)
> 
> indx = eig.argsort()
> eig = eig.take(indx)
> val = val.take(indx, axis=1)
> master = zip(eig, val.T)

Thank you, that worked.
http://www.ogre.nu/wp/?p=1978

I refined it slightly:

val,vec = numpy.linalg.eig(adj)
indx = val.argsort()[-4:-1]
val = val.take(indx)
vec = vec.take(indx, axis=1)
master = zip(val, vec.T)


-- 
Anton Sherwood, http://www.ogre.nu/
"How'd ya like to climb this high *without* no mountain?" --Porky Pine




More information about the NumPy-Discussion mailing list