Indexing eines Array von einer anderen in numpy

Question

Angenommen, ich habe eine Matrix A mit einigen beliebigen Werten:

array([[ 2, 4, 5, 3], 
     [ 1, 6, 8, 9], 
     [ 8, 7, 0, 2]]) 

und eine Matrix B die Indizes der Elemente in A enthält:

array([[0, 0, 1, 2], 
     [0, 3, 2, 1], 
     [3, 2, 1, 0]]) 

Wie wähle ich Werte von A spitz von B, dh:

A[B] = [[2, 2, 4, 5], 
     [1, 9, 8, 6], 
     [2, 0, 7, 8]] 

Answer 1

können Sie verwenden NumPy's advanced indexing -

m,n = A.shape 
out = np.take(A,B + n*np.arange(m)[:,None]) 

Probelauf - -

A[np.arange(A.shape[0])[:,None],B] 

One auch linear indexing können

In [40]: A 
Out[40]: 
array([[2, 4, 5, 3], 
     [1, 6, 8, 9], 
     [8, 7, 0, 2]]) 

In [41]: B 
Out[41]: 
array([[0, 0, 1, 2], 
     [0, 3, 2, 1], 
     [3, 2, 1, 0]]) 

In [42]: A[np.arange(A.shape[0])[:,None],B] 
Out[42]: 
array([[2, 2, 4, 5], 
     [1, 9, 8, 6], 
     [2, 0, 7, 8]]) 

In [43]: m,n = A.shape 

In [44]: np.take(A,B + n*np.arange(m)[:,None]) 
Out[44]: 
array([[2, 2, 4, 5], 
     [1, 9, 8, 6], 
     [2, 0, 7, 8]]) 

Answer 2

Es folgt der Lösung unter Verwendung for Schleife:

outlist = [] 
for i in range(len(B)): 
    lst = []  
    for j in range(len(B[i])): 
     lst.append(A[i][B[i][j]]) 
    outlist.append(lst) 
outarray = np.asarray(outlist) 
print(outarray) 

Above können auch in prägnanter Liste Verständnis Form geschrieben werden:

outlist = [ [A[i][B[i][j]] for j in range(len(B[i]))] 
       for i in range(len(B)) ] 
outarray = np.asarray(outlist) 
print(outarray) 

Ausgang:

[[2 2 4 5] 
[1 9 8 6] 
[2 0 7 8]]