dos matrices que definen coordenadas 2d, como índices de matriz, en matlab/octave

Question

Tengo una matriz 2D, llámala 'A'. Tengo otras dos matrices en 2D, llámalas 'ix' e 'iy'. Me gustaría crear una matriz de salida cuyos elementos son los elementos de A en los pares de índices proporcionados por x_idx y y_idx. Puedo hacer esto con un ciclo de la siguiente manera:

for i=1:nx 
    for j=1:ny 
     output(i,j) = A(ix(i,j),iy(i,j)); 
    end 
end 

¿Cómo puedo hacer esto sin el bucle? Si hago output = A (ix, iy), obtengo el valor de A en todo el rango de (ix) X (iy).

Gracias, Jason

Answer 1

Una forma más rápida es utilizar indexación lineal directamente sin llamar SUB2IND:

output = A(size(A,1)*(iy-1) + ix) 

... pensar en la matriz A como una matriz de 1D (orden por columnas)

Answer 2

Este es el método de una línea que no es muy eficiente para grandes matrices

reshape(diag(A(ix(:),iy(:))),[ny nx]) 

Un método más claro y más eficiente sería utilizar sub2ind. He incorporado el comentario de puaj para situaciones (como la suya) cuando ix y iy tienen el mismo número de elementos:

newA = A(sub2ind(size(A),ix,iy)); 

Además, no hay que confundir x y y para ij y en notación - j y x general Representar columnas y i y y representan filas.