Доступ к списку записей в MATLAB

Question

У меня есть огромная матрица MxN матрицу, скажем, A=rand([M,N]); и вектор индекса с N целочисленных значений между 1 и M, скажем, RandomIndex = randi(M,[1,N]);.

Теперь я хотел бы, чтобы сгенерировать вектор-строку с элементами

result = [A(RandomIndex(1),1), A(RandomIndex(2),2), ..., A(RandomIndex(N),N)] 

Что бы быть эффективным способом сделать это? Это должна быть очень дешевая операция, но все мои реализации медленны. Я не думаю, что в Matlab есть обозначение, чтобы сделать это прямо, не так ли?

Самый быстрый вариант до сих пор является

indexFunction = @(r,c) A(r,c); 
result = cell2mat(arrayfun(indexFunction,RandomIndex,1:N,'UniformOutput',false)); 

Есть ли более эффективный способ?

Answer 1

Использование sub2ind

A(sub2ind(size(A), RandomIndex, 1:N)) 

sub2ind преобразует строк и столбцов индексов, данные RandomIndex и 1:N линейных индексов на основе size(A), которые затем можно использовать для индекса A непосредственно.

---

Другой способ сделать это состоит в использовании RandomIndex и 1:N вернуть NxN матрицу, а затем взять диагональ это с diag

diag(A(RandomIndex, 1:N)).' 

Примечание: .' используется для преобразования строки вектор возвращается diag к вектору столбца.

Answer 2

M=10;N=50; 
A=rand([M,N]); 
RandomIndex = randi(M,[1,N]); 
out = zeros(1,numel(RandomIndex)); 
for ii = 1:numel(RandomIndex) 
    out(ii)=A(RandomIndex(ii),ii); 
end 

Answer 3

Используйте ваш индекс напрямую.

M = 100;N=100; 
A = rand(M,N); 
% get a random index that can be as large as your matrix 
% 10 rows by 1 column 
idx = randi(numel(A), 10,1); 
t = A(idx); 

Answer 4

Другой подход заключается в использовании sparse и логическую индексацию:

M = sparse(RandomIndex, 1:N, 1) == 1; 
out = A(M); 

Первая строка кода создает логическую матрицу, где существует только один true значение, установленное в каждом столбце. Это определяется каждым значением RandomIndex. Мы преобразуем это в матрицу logical, а затем индексируем в вашу матрицу, чтобы получить конечный случайный вектор.