在 Matlab 中挑选给定指定列的矩阵条目，无需 for 循环

Question

可能的重复：
MATLAB 索引问题
如何在Matlab中从矩阵中提取非垂直列< /a>

我觉得应该有一个简单的方法来做我想做的事，但我无法弄清楚这一点。

输入：一个 nx t 实数矩阵 M 和一个 nx 1 向量 I索引

OUTPUT：一个 nx 1 向量 P，使得 P(i) = M( i, I(i) )< /code>

很明显如何使用for 循环，但这是 Matlab，并且 n 很大。有没有办法向量化这个问题并避免 for 循环？

Answer 1

这是使用线性索引的简单、快速、矢量化解决方案。

indx = (1:n)' + (I-1)*n; %'
P=M(indx);

---

例子：

M = randi(10,[3,4]);     %# test matrix

M =

     9    10     3    10
    10     7     6     2
     2     1    10    10

n = size(M,1);
I = [3,1,4]';            %'# index vector
indx = (1:n)' + (I-1)*n; %'
P = M(indx)

P =

 3
10
10