Comprensión de la lista de Matlab

Question

¿Puedo escribir lo siguiente en un trazador de líneas único?

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

Supongamos que la función foo no maneja las matrices como se esperaba. En mi caso, foo devuelve un escalar incluso cuando se da una matriz como argumento.

En Python, por ejemplo, se vería así:

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

Answer 1

arrayfun es lo que necesita. Por ejemplo:

res = arrayfun(@foo, x) 

Desde foo siempre devuelve un escalar, el anterior funcionará y res También habrá un vector de las mismas dimensiones que x. Si foo devuelve salida de longitud variable, tendrá que configurar 'UniformOutput' en false o 0 en la llamada al arrayfun. La salida será entonces una matriz cell.

Answer 2

sólo para añadir a la buena respuesta de @yoda, en lugar de utilizar UniformOutput, también se puede utilizar {} paréntesis:

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

También, en algunas ocasiones, se foo ya vectorizado.

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