Discussion :

[ol9245]

Bonjour,

J'ai un tableau I, (I comme index), qui me dit quelle ligne je dois extraire dans chaque colonne d'une matrice M. ci-dessous mon code (qui marche).

Code :

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

1
2
3
4
5
6
7
8
9
M = magic(5)
I = [1, 1, 2, 5, 4] ;
R = zeros(1,5) ;
 
for i=1:5
    R(i) = M(I(i), i) ;
end
 
disp (R) ; % 17    24     7     2     3

Je me demande s'il est possible de faire la même chose directement, sans boucle. (souci d'optimisation pour code qui s'annonce gourmand)

[Jerome Briot]

Il suffit de passer par l'indexage linéaire => http://briot-jerome.developpez.com/m...sgestion/#LIII

Code :

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

1
2
3
4
5
 
[r,c] = size(M);
idx = (0:c-1)*r+I;
 
R = M(idx)

ou en utilisant la fonction SUB2IND :

Code :

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

1
2
3
4
[r,c] = size(M);
idx = sub2ind([r c],I,1:c);
 
R = M(idx)

La seconde solution sera moins rapide mais plus robuste à cause de la batterie de tests effectués au début de la fonction SUB2IND

[duf42]

Moi j'en ai un aussi mais il semblerait que ca ne fait pas gagner de temps

Code :

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

1
2
 
R=diag(M(I,:));

[ol9245]

J'y crois pas Vous êtes vraiment les meilleurs !!!

adressage linéaire : la seule chose que je savais faire (mais zut)
diag : c'est beau ! rien que pour ça, merci !
sub2ind, c'est ce qu'il me faut.

Bravo à vous deux et ...