MATLAB索引问题

Question

例如，我有一个矩阵

A = [ 1 2 3; 4 5 6; 7 8 9] ;

和一个大小为 1x3 的向量，它指定每行中的哪个元素是我要查找的元素 - 即如果

vector = [ 1 2 1 ]

那么所需的输出是

[ 1 5 7 ]

因为 1 是第一个第 1 行中的元素，5 是第 2 行中的第 2 个元素，7 是第 3 行中的第 1 个元素。

我该如何实现这一目标？找不到内置函数来执行此操作，这让我感到惊讶。

Answer 1

MATLAB 提供 SUB2IND 函数将行/列下标转换为线性索引：

>> A = [1 2 3; 4 5 6; 7 8 9];
>> idx = sub2ind(size(A),1:3,[1 2 1]);  %# rows: [1 2 3], cols: [1 2 1]
>> A(idx)
     1     5     7

Answer 2

首先，Matlab中的索引是从上到下的。
所以在你的情况下 A[1] = 1 , A[2] = 4 , A[3] = 7

这就是说，处理 A' 会更容易，因为它有点微不足道。

B = A';

B((vector + [0:2].* 3))

Answer 3

这有点难看，但是 diag(A(1:3,[1 2 1])) 可以解决问题。

Answer 4

这是 Yochai 的答案的变体，但没有转置（这基本上也是SUB2IND 在 Amro 的回答）：

 output = A((1:3)+3.*(vector-1));

或者对于任意大小的数组 A：

 nRows = size(A,1);
 output = A((1:nRows)+nRows.*(vector-1));