我在对numpy数组进行排序并获取排序数组的索引时遇到了一些问题,但是将索引应用于原始数组并不能达到我的预期。因此,这是我正在做的测试用例:
import numpy as np
# Two 3x3 matrices
x = np.random.rand(2, 3, 3)
# Perform some decomposition (Never mind the matrices are not hermitian...)
evals, evecs = np.linalg.eigh(x)
# evals has shape (2, 3), evecs has shape (2, 3, 3)
indices = evals.argsort(axis=1)[..., ::-1] # Do descending sort
# Now I want to apply the index to evals.
evals = evals[:, indices]
我没有获得(2,3)阵列,而是获得了(2,3,3)阵列回到复制行的位置。就像是:
array([[[ 1.15628047, 0.16853886, -0.28607138],
[ 1.15628047, 0.16853886, -0.28607138]],
[[ 2.4311532 , -0.00754817, -0.24086572],
[ 2.4311532 , -0.00754817, -0.24086572]]])
我不确定为什么会这样。将不胜感激。
最佳答案
这应该工作:
import numpy as np
idx0 = np.arange(evals.shape[0])[:,np.newaxis]
idx1 = evals.argsort(1)[...,::-1]
evals[idx0,idx1]
这会按照降序分别对每一行进行排序。
编辑:
在这种情况下,您需要
(idx0,idx1)
进一步处理特征向量evecs
。如果不是这种情况,那就很简单evals.sort()
evals = evals[:,::-1]