numpy : 색인 생성 관련 문제

Question

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) 배열, I는 (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]]]) 

왜 그런지는 잘 모르겠습니다. 어떤 도움을 주시면 감사하겠습니다.

Answer 1

이 작동합니다 :

import numpy as np 
idx0 = np.arange(evals.shape[0])[:,np.newaxis] 
idx1 = evals.argsort(1)[...,::-1] 
evals[idx0,idx1] 

이 순서를 감소시켜, 개별적으로, 각 행을 정렬합니다.

편집 : 이 경우 고유 벡터 evecs을 추가로 처리하려면 (idx0,idx1)이 필요합니다. 이 경우 아니었다면,

evals.sort() 
evals = evals[:,::-1]