Discussion :

[nant44]

Bonjour,

J'ai une matrice de données
A de dimension 100 x N (N >>>> 1)
et une matrice d'indices B de dimension 10 x N
Question :
Existe-t-il une façon d'extraire la matrice de données 10 x N sans faire une boucle for sur N ?

Autrement dit, peut-on remplacer la boucle for (dans le code ci-dessous) par une ligne ?

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
        N = 20; % ou 2000000
        A = randn(10,N);
        B = randi([1 10],5,N);
        C = zeros(5,N);
        for n = 1:N
            C(:,n) = A(B(:,n),n);
        end

merci

[Invité]

Bonjour,

Comme ceci :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

C = A( sub2ind(size(A), B, repmat(1:N, 5, 1) ) );

[nant44]

Exact, merci.