別の配列のインデックスを持つ2次元配列から値を抽出する（ループなし）

Question

私は配列[2; 3]と行列[ 1 3 4 5; 2 4 9 2]を持っています。 ここでは、最初の行から2番目の要素を抽出し、2番目の行から3番目の要素を抽出して、[3 ; 9]を取得します。私はそれをループでやっていましたが、もっと大きな配列で作業しているので、これらを避けたいと思います。

Answer 1

あなたはlinear indexに（その行の添字と一緒に）列の添字のそれぞれを変換するためにsub2indを使用して、あなたの行列にインデックスにことを使用することができます。

A = [1 3 4 5; 2 4 9 2]; 
cols = [2; 3]; 

% Compute the linear index using sub2ind 
inds = sub2ind(size(A), (1:numel(cols)).', cols); 

B = A(inds) 
% 3 
% 9 

代わりに、あなたはdiag機能を利用することによりsub2ind

B = A((cols - 1) * size(A, 1) + (1:numel(cols)).'); 
% 3 
% 9 

Answer 2

よりもパフォーマンスになるだろう自分線形インデックスを計算することができ、あなたは、エレガントな1行の解を得ることができます。

A = [1 3 4 5; 2 4 9 2]; 
cols = [2; 3]; 

B = diag(A(:,cols)) 
% 3 
% 9 

---

ここにあるのはdiag(A(:,cols))が行う：

1. A(:,cols)がカラムを、colsAの列を選択Aのcols(k)は、[3 4; 4 9]を与える列に対応A(:,cols)のk。
2. diagは、kのk第11番目の対角要素がA(:,cols)であるA(k,cols(k))を返します。この行列の対角エントリを返します。