如何從matlab中矩陣的每一列中選擇一個元素？

Question

a = [1 1 1; 2 2 2; 3 3 3]; 

b = [1 2 3]; 

如何調用一個函數來獲得矢量v[i] = a[b[i],i]？

Answer 1

v = a(sub2ind(size(a), b, 1:length(b))) 

sub2ind將下標轉換爲單個索引。

Answer 2

要嘗試的另一件事，保持非常接近您的描述，您可以使用arrayfun函數。

首先定義一個函數，將值x映射到期望的輸出。

fn = @(x) a(b(x), x); 

然後調用在所述的i向量中的每個輸入端，功能。

i = 1:3; 
v = arrayfun(fn, i); 

或者，這可以在一個單一的線來完成，以明顯的方式：

v = arrayfun(@(x) a(b(x), x), 1:3); 

這arrayfun簡直是簡寫以下循環：

for ixLoop = 1:3 
    v(ixLoop) = a(b(ixLoop),ixLoop); 
end 

的相關功能arrayfun,cellfun和structfun具有類似的用途，並且奇怪地賦予權力。 This Joel article說服我是一個信徒。