matlab 判断元素索引,在MATLAB中的spfun，cellfun，arrayfun等中检索元素索引

女爷i 2022-08-30 09:10 145阅读 0赞

![Image 1][]

Is there any way to retrieve the index of the element on which a function called by cellfun, arrayfun or spfun acts? (i.e. retrieve the index of the element within the scope of the function).

For the sake of simplicity, imagine I have the following toy example:

S = spdiags(\[1:4\]',0,4,4)

f = spfun(@(x) 2\*x,S)

which builds a 4x4 sparse diagonal matrix and then multiplies each element by 2.

And say that now, instead of multiplying each element by the constant number 2, I would like to multiply it by the index the element has in the original matrix, i.e. assuming that linear\_index holds the index for each element, it would be something like this:

S = spdiags(\[1:4\]',0,4,4)

f = spfun(@(x) linear\_index\*x,S)

However, note the code above does not work (linear\_index is undeclared).

This question is partially motivated by the fact that blocproc gives you access to block\_struct.location, which one could argue references the location (~index) of the current element within the full object (an image in this case):

block\_struct.location: A two-element vector, \[row col\], that specifies

the position of the first pixel (minimum-row, minimum-column) of the

block data in the input image.

解决方案

No, but you can supply the linear index as extra argument.

Both cellfun and arrayfun accept multiple input arrays. Thus, with e.g. arrayfun, you can write

a = \[1 1 2 2\];

lin\_idx = 1:4;

out = arrayfun(@(x,y)x\*y,a,lin\_idx);

This doesn't work with spfun, unfortunately, since it only accepts a single input (the sparse array).

You can possibly use arrayfun instead, like so:

S = spdiags(\[1:4\]',0,4,4);