matlab - ループを使用せずにマトリックスからエントリを選択する

Question

私は2つの行列Aとを持っています。BどちらもNx3行列です。

現在、以下を使用して、行列Aの各行の最大値とインデックスを取得しています。

[maxA, idx] = max(A, [], 2)

idx(j)行の最大値が含まれている列を示しますj。ここで、行列Bから同じ位置を選択したいと思います。

私は現在、ループを使用してこれを実装しています：

for j = 1:numel(idx)
   maxB(j) = B(j, idx(j))
end

私の現在の実装は十分に高速ですが、不要なループを避けたいので、ループなしでこれを表現する方法はありますか？

Answer 1

線形インデックスのベクトルを作成できます（BはAと同じサイズであると思います）。

vec_indices = sub2ind(size(A), 1:numel(idx), idx);

次に、そのベクトルをルックアップに直接使用できます。

maxB = B(vec_indices)

Answer 2

単一次元のインデックスをマトリックスに構築し、その方法で取得できます。matlabのすべての多次元行列に対応できます。

使用できます

maxB = B(sub2ind([1:length(idx)]',idx(:)));

Answer 3

一行で：

maxB = B(A == max(A, [], 2) * ones(1, 3));

しかし、これは安全ではありません。のすべての行で一意の値を想定していますA。