python numpy排序特徵值

Question

我正在使用linalg.eig（A）來獲取矩陣的特徵值和特徵向量。有沒有簡單的方法來排序這些特徵值（和相關向量）？

Answer 1

您想使用NumPy sort()和argsort()函數。 argsort()返回數組排序需要指數的排列，所以如果你想通過特徵值大小進行排序（用於與NumPy陣列的標準排序似乎是最小的到最大的），你可以這樣做：

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通常會返回複雜的值。您可能需要考慮使用np.sort_complex(eig_vals)。