Using vector to access multidimensional array or passing vector as multiple inputs
1 vue (au cours des 30 derniers jours)
Afficher commentaires plus anciens
Hello,
I would like to access multidimensional array using vector like:
n=2
arraySizes=[n*ones(1,2),3*ones(1,3)]
A=ones(arraySizes)
ind=[3,2,1,1,2]
A(ind) % I would like it, if output was a scalar.
or to convert that vector of indices to linear indice:
indN= sub2ind(arraySizes,ind)
A(indN)
I know that it is possible to use cell and {:}, but num2cell and mat2cell aren't supported in MEX creation, so it's also not a solution I look for (and for loop copying num array to cell is very slow).
indCell=num2cell(ind)
indN= sub2ind(arraySizes,indCell{:})
A(indN)
I don't want just use hard indexing, because n isn't constant, and switch case with hundreds hard coded options doesn't seem optimal (but maybe it would be the fastest method then I will).
A(ind(1),ind(2),ind(3),ind(4),ind(5))
My current solution is showed below, but I`m looking for faster approch.
indNum=(ind-ones(1,length(arraySizes)))*((cumprod(arraySizes))./arraySizes).'+1
A(indNum)
Thank you for your time,
Matt
2 commentaires
Rik
le 17 Juin 2022
How is ind created? It might be easier to make sure it is a cell from the start. Then you can use A(ind{:}).
Réponses (1)
Chetan
le 5 Jan 2024
I understand you're looking to access elements in a multidimensional array for MEX file creation without using `num2cell` or `mat2cell`. Your current method of converting subscript indices to a linear index is correct and efficient for this purpose.
Here's a concise way to calculate the linear index:
n = 2;
arraySizes = [n*ones(1,2), 3*ones(1,3)];
A = ones(arraySizes);
ind = [3, 2, 1, 1, 2];
% Calculate cumulative product of array sizes
cumprodSizes = [1, cumprod(arraySizes(1:end-1))];
% Compute linear index and access the element
linearIndex = sum((ind - 1) .* cumprodSizes) + 1;
elementValue = A(linearIndex);
This code calculates the linear index by considering the array's dimensions and then accesses the element directly. It's a MEX-friendly approach and scales with different array dimensions.
Make sure the subscript indices in `ind` are valid for the array's size before indexing.
For more information about array indexing and linear indexing in MATLAB, you might find these resources helpful:
- Array Indexing: https://www.mathworks.com/help/matlab/math/array-indexing.html
- Linear Indexing: https://www.mathworks.com/help/matlab/math/matrix-indexing.html
Hope it helps
Best regards,
Chetan Verma
0 commentaires
Voir également
Catégories
En savoir plus sur Loops and Conditional Statements dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!