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)。