如何从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说服我是一个信徒。