Call function for all elements in an array

问题

Let's say I have a function, like:

function [result] = Square( x )
    result = x * x;
end

And I have an array like the following,

x = 0:0.1:1;

I want to have an y array, which stores the squares of x's using my Square function. Sure, one way would be the following,

y = zeros(1,10);

for i = 1:10
    y(i) = Square(x(i));
end

However, I guess there should be a more elegant way of doing it. I tried some of my insights and made some search, however couldn't find any solution. Any suggestions?

回答1:

For the example you give:

y = x.^2;   % or
y = x.*x;

in which .* and .^ are the element-wise versions of * and ^. This is the simplest, fastest way there is.

More general:

y = arrayfun(@Square, x);

which can be elegant, but it's usually pretty slow compared to

y = zeros(size(x));
for ii = 1:numel(x)
    y(ii) = Square(x(ii)); end

I'd actually advise to stay away from arrayfun until profiling has showed that it is faster than a plain loop. Which will be seldom, if ever.

In new Matlab versions (R2008 and up), the JIT accelerates loops so effectively that things like arrayfun might actually disappear in a future release.

As an aside: note that I've used ii instead of i as the loop variable. In Matlab, i and j are built-in names for the imaginary unit. If you use it as a variable name, you'll lose some performance due to the necessary name resolution required. Using anything other than i or j will prevent that.

回答2:

You want arrayfun.

arrayfun(@Square, x)

See help arrayfun

(tested only in GNU Octave, I do not have MATLAB)

回答3:

Have you considered the element-by-element operator .*?

See the documentation for arithmetic operators.

回答4:

I am going to assume that you will not be doing something as simple as a square operation and what you are trying to do is not already vectorised in MATLAB.

It is better to call the function once, and do the loop in the function. As the number of elements increase, you will notice significant increase in operation time.

Let our functions be:

function result = getSquare(x)
    result = x*x; % I did not use .* on purpose
end

function result = getSquareVec(x)
result = zeros(1,numel(x));
    for idx = 1:numel(x)
        result(:,idx) = x(idx)*x(idx);
    end
end

And let's call them from a script:

y = 1:10000;
tic;
for idx = 1:numel(y)
    res = getSquare(y(idx));
end
toc

tic;
    res = getSquareVec(y);
toc

I ran the code a couple of times and turns out calling the function only once is at least twice as fast.

Elapsed time is 0.020524 seconds.
Elapsed time is 0.008560 seconds.

Elapsed time is 0.019019 seconds.
Elapsed time is 0.007661 seconds.

Elapsed time is 0.022532 seconds.
Elapsed time is 0.006731 seconds.

Elapsed time is 0.023051 seconds.
Elapsed time is 0.005951 seconds.

来源：https://stackoverflow.com/questions/13816757/call-function-for-all-elements-in-an-array

标签

function

matlab

indexing