pick elements from a 3d array, based on an indexing matrix
17 views (last 30 days)
Show older comments
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 Comments
Accepted Answer
Stephen23
on 26 Jan 2021
Edited: Stephen23
on 26 Jan 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 Comments
Stephen23
on 27 Jan 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.
More Answers (0)
See Also
Categories
Find more on Matrices and Arrays in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!