Matlab列表理解

Question

我可以在一個班輪中寫下以下內容嗎？

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

假設foo函數沒有按預期處理數組。在我的情況下，即使給出一個數組作爲參數，foo也會返回一個標量。

在Python中，例如，它應該是這樣的：

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

Answer 1

arrayfun是你所需要的。例如：

res = arrayfun(@foo, x) 

foo由於總是返回一個標量，上述將工作和res也將是相同的尺寸x的向量。如果foo返回可變長度輸出，則必須在arrayfun的調用中將'UniformOutput'設置爲false或0。輸出將是一個cell陣列。

Answer 2

只需添加到@yoda的很好的答案，而不是使用UniformOutput，你也可以使用{}括號：

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

而且，在某些場合，foo已經被量化。

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