在python中使用numpy.linalg.eig之后，对特征值和相关的特征向量进行排序

Question

我正在使用numpy.linalg.eig获得特征值和特征向量的列表：

A = someMatrixArray
from numpy.linalg import eig as eigenValuesAndVectors

solution = eigenValuesAndVectors(A)

eigenValues = solution[0]
eigenVectors = solution[1]

我想对我的特征值进行排序（例如，从最低到最高），以某种方式知道排序后相关的特征向量是什么。

我没有找到任何使用python函数的方法。有什么简单的方法还是我必须对我的排序版本进行编码？

Answer 1

使用numpy.argsort。它返回一个索引，该索引将用于对数组进行排序。

import numpy as np
import numpy.linalg as linalg

A = np.random.random((3,3))
eigenValues, eigenVectors = linalg.eig(A)

idx = eigenValues.argsort()[::-1]   
eigenValues = eigenValues[idx]
eigenVectors = eigenVectors[:,idx]

如果特征值是复数，则排序顺序是按字典顺序排列（也就是说，复数首先根据其实部进行排序，而其虚部则打破了联系）。