在给定每行的列索引的情况下，如何从矩阵的每行中选择一个元素

Question

例如，如果矩阵是

A = [11 22 33 11; ...
44 55 66 34; ...
67 45 33 22]

然后我想从每一行中选择

col_idx = [2 4 1]    

因此结果将是

ans = 22
34
67

其他类似的问题基于R或Python。然而，我正在寻找一个基于MATLAB的答案。如有任何帮助，我们将不胜感激。

Answer 1

我会使用sub2ind：

A = [11 22 33 11; 44 55 66 34; 67 45 33 22];
col_idx = [2 4 1]
row_idx = 1:size(A, 1)
A(sub2ind(size(A), row_idx, col_idx))

输出：

col_idx =
2   4   1
row_idx =
1   2   3
ans =
22   34   67

希望能有所帮助！