python numpy sort wartości własne

Question

Używam linalg.eig (A), aby uzyskać wartości własne i wektory własne macierzy. Czy istnieje prosty sposób sortowania tych wartości własnych (i powiązanych wektorów) w kolejności?

Answer 1

Chcesz używać funkcji NumPy sort() i argsort(). argsort() zwraca permutacji wskaźników potrzebnych do sortowania tablicy, więc jeśli chcesz, aby posortować według wartości własnej wielkości (średnia sortowania tablic numpy wydaje się być najmniejszym do największych), można zrobić:

import numpy as np 

A = np.asarray([[1,2,3],[4,5,6],[7,8,9]]) 
eig_vals, eig_vecs = np.linalg.eig(A) 

eig_vals_sorted = np.sort(eig_vals) 
eig_vecs_sorted = eig_vecs[:, eig_vals.argsort()] 


# Alternatively, to avoid making new arrays 
# do this: 

sort_perm = eig_vals.argsort() 

eig_vals.sort()  # <-- This sorts the list in place. 
eig_vecs = eig_vecs[:, sort_perm] 

Answer 2

np.linalg.eig często zwraca wartości zespolone. Możesz rozważyć użycie np.sort_complex(eig_vals).