在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]

如果特征值是复数，the sort order is lexicographic（也就是说，复数首先根据其实部进行排序，并由其虚部断开连接）。