How can I access the elements of one array with another array?

I'd like to assign elements to an array in the following manner:
a = zeros(2,2,2);
ix=1;
iy=1;
b = [1,1];
a([ix,iy],:) = b(:)
Where ix and iy represent the position in the 1st and 2nd directions, a in it's 3rd dimension and b are the same length length. For my application ix and iy really need to be in the same array and the method needs to be applicable to N-dimensions. Is there a way to do such an assignment?

 采纳的回答

Stephen23
Stephen23 2018-12-11
编辑:Stephen23 2018-12-11
This is really very easy once you put those indices into a cell array:
C = {ix,iy,...,iN,':'};
a(C{:}) = b
Given a numeric vector of indices V:
V = [ix,iy,...,iN];
C = num2cell(V);
a(C{:},':') = b;
or
a(C{:},:) = b;

更多回答(2 个)

a = zeros(2,2,2);
ix=1;
iy=1;
b = [1,1];
a(ix,iy,:) = b

1 个评论

Sorry, but I would like to be able to extend this to N-dimensions without writing:
a(ix,iy,...,iN,:) = b
The indices [ix,iy,...,iN] are generated programmatically the length varies.

请先登录,再进行评论。

I'm going to assume that [ix, iy] is a 2D array, not a vector as in your example, otherwise your problem is trivially solved with:
indices = [ix, iy];
a(indices(1), indices(2), :) = b(:);
If it's a vector, you will have to convert your Nd subscripts into linear indices with sub2ind. Unfortunately, that means you can't use the colon operator. You have to explicitly list the dimensions.
a = zeros(2, 2, 2);
indices = [1 1;1 2;2 1];
b = [1 2; 3 4; 5 6] %shape of b doesn't matter
linindices = sub2ind(size(a), repmat(indices(:, 1), 1, size(a, 3)), repmat(indices(:, 2), 1, size(a, 3)), repmat(1:size(a, 3), size(indices, 1), 1))
a(linindices) = b

2 个评论

[ix, iy] is a vector, but the size of 'a' varies programmatically as do the indices so the vector could look like indices [ix,iy,...,iN]. And I'd like to assign values as follows
a(ix,iy,...,iN,:) = b
I'll investigate linear indices in the meantime.
Ah ok, In that case, see Stephen's answer

请先登录,再进行评论。

类别

帮助中心File Exchange 中查找有关 Resizing and Reshaping Matrices 的更多信息

产品

版本

R2018b

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by