pick elements from a 3d array, based on an indexing matrix
显示 更早的评论
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?
采纳的回答
% 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 个评论
Bernhard
2021-1-26
Stephen23
2021-1-27
@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.
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 Matrices and Arrays 的更多信息
Translated by 
