Сканирование каждой колонки без петли в MATLAB

Question

I = [2 1]; 
A = [7 11 
    8 17]; 

Как выбрать правильное значение в A без for? I - вектор индексов для каждого столбца в A. То есть I(1) is 8 и I(2) = 11.

Answer 1

Используйте sub2ind для генерации линейных индексов, основанных на правильных координатах строк и столбцов, затем используйте их для индексации в A. В этом случае I выбирает правый ряд, и вы хотите, чтобы выбрать только один элемент для каждого столбца от первого до последнего:

ind = sub2ind(size(A), I, 1:numel(I)); 
out = A(ind); 

Пример

>> I = [2 1]; 
>> A = [7 11 
     8 17]; 
>> ind = sub2ind(size(A), I, 1:numel(I)); 
>> out = A(ind); 
>> out 

out = 

    8 11