pick elements from a 3d array, based on an indexing matrix
15 visualizaciones (últimos 30 días)
Mostrar comentarios más antiguos
Bernhard
el 26 de En. de 2021
Comentada: Stephen23
el 27 de En. de 2021
I have a 3d array A with nr rows, nc columns, and np pages, and an nr x nc indexing matrix I with integer elements between 1 and np. I want to create a matrix B with elements picked from A, where for each position (r,c) the element is picked from the page of A given by the value of I at that position.
A straightforward way to do this would be the double loop:
B = zeros(nr,nc);
for r = 1:nr
for c = 1:nc
p = I(r,c);
B(r,c) = A(r,c,p);
end
end
Is there a way to accelerate this by vectorization for large arrays?
0 comentarios
Respuesta aceptada
Stephen23
el 26 de En. de 2021
Editada: Stephen23
el 26 de En. de 2021
% fake data:
nr = 5;
nc = 7;
np = 3;
A = randi(9,nr,nc,np);
I = randi(np,nr,nc)
% your looped approach:
B = zeros(nr,nc);
for r = 1:nr
for c = 1:nc
p = I(r,c);
B(r,c) = A(r,c,p);
end
end
B
% SUB2IND approach:
[xr,xc] = ndgrid(1:nr,1:nc);
X = sub2ind(size(A),xr,xc,I);
C = A(X)
2 comentarios
Stephen23
el 27 de En. de 2021
@Bernhard: if the speed of this operation is very critical to your algorithm, then take a look inside the sub2ind file. The sub2ind function is generic, for any number of dimensions, but with a little effort you can write your own version that is hardcoded for the number of dimensions of your data and that does not require ndgrid beforehand. Whether it is worth the effort depends on your situation.
Más respuestas (0)
Ver también
Categorías
Más información sobre Matrices and Arrays en Help Center y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!