CODEWITHSUNDEEP
matlablist-comprehension
Lucy Brennan
Lucy Brennan

Reputation: 595

Matlab list comprehension

Can I write the following in a one liner?

x = [1,3,5,7,9]
res = zeros(1,size(x,2));
for i=1:size(x,2);
    res(i) = foo(x(i));
end;

Assume that the foo function does not handle arrays as expected. In my case foo returns a scalar even when giving an array as argument.

In Python, for instance, it would look like this:

x = [1,3,5,7,9]
res = [foo(y) for y in x]

Upvotes: 16

Views: 32123

Answers (2)

abcd
abcd

Reputation: 42225

arrayfun is what you need. For example:

res = arrayfun(@foo, x)

Since foo always returns a scalar, the above will work and res will also be a vector of the same dimensions as x. If foo returns variable length output, then you will have to set 'UniformOutput' to false or 0 in the call to arrayfun. The output will then be a cell array.

Upvotes: 18

Andrey Rubshtein
Andrey Rubshtein

Reputation: 20915

Just to add to the good answer of @yoda, instead of using UniformOutput, you can also use {} brackets:

res = arrayfun(@(t){foo(t)}, x)

Also, in some occasions, foo is already vectorized.

x = 1:10;
foo = @(t)(power(t,2));
res = foo(x);

Upvotes: 7

Related Questions