Matlab compréhension de la liste

Puis-je écrire la suite en une seule ligne?

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

Supposer que les foo fonction ne gère pas les tableaux comme prévu. Dans mon cas, toto retourne un scalaire, même lorsque vous donnez un tableau comme argument.

En Python, par exemple, il devrait ressembler à ceci:

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

source d'informationauteur Lucy Brennan

  1. 10

    arrayfun est ce que vous avez besoin. Par exemple:

    res = arrayfun(@foo, x)

    Depuis foo toujours retourne un scalaire, le ci-dessus ne fonctionne et res sera également un vecteur de même dimension que x. Si foo retourne la longueur variable de sortie, alors vous aurez à définir 'UniformOutput' à false ou 0 dans l'appel à arrayfun. La sortie sera alors un cell tableau.

  2. 6

    Juste pour ajouter à la bonne réponse de @yoda, au lieu d'utiliser UniformOutputvous pouvez également utiliser {} parenthèses:

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

    Aussi, en certaines occasions, foo est déjà vectorisé.

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