Python中使用numpy.linalg.eig后对特征值和相关特征向量进行排序

使用numpy.linalg.eig获得特征值和特征向量的列表:

A = someMatrixArray
from numpy.linalg import eig as eigenValuesAndVectors

solution = eigenValuesAndVectors(A)

eigenValues = solution[0]
eigenVectors = solution[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(也就是说,复数将首先根据其实部进行排序,并用虚部断开连接)。

你可能感兴趣的:(Python语言专栏,数学大类专栏,python,算法,开发语言)