Existe-t-il une manière vectorisée d'opérer sur un nombre différent de valeurs par colonne dans MATLAB?

Question

Dans MATLAB, existe-t-il un moyen plus concis de gérer l'indexation conditionnelle discrète par colonne plutôt que d'utiliser une boucle for? Voici mon code:

x=[1 2 3;4 5 6;7 8 9]; 
w=[5 3 2]; 

q=zeros(3,1); 
for i = 1:3 
    q(i)=mean(x(x(:,i)>w(i),i)); 
end 
q 

Mon but est de prendre la moyenne de la partie supérieure x% d'un ensemble de valeurs pour chaque colonne. Le code ci-dessus fonctionne, mais je me demande simplement s'il existe une façon plus concise de le faire?

Answer 1

Vous avez mentionné que vous utilisiez la fonction PRCTILE, qui indiquerait que vous avez accès à Statistics Toolbox. Cela vous donne encore une autre option pour la façon dont vous pourriez résoudre votre problème, en utilisant la fonction NANMEAN. Dans le code suivant, toutes les entrées de x inférieure ou égale au seuil w pour une colonne sont mis à NaN en utilisant BSXFUN, la moyenne de chaque colonne est calculée avec NANMEAN:

x(bsxfun(@le,x,w)) = nan; 
q = nanmean(x); 

Answer 2

Je ne connais aucun moyen d'indexer les colonnes comme vous le souhaitez. Cela peut être plus rapide qu'une boucle for, mais cela crée aussi une matrice y qui a la taille de x.

x=[1 2 3;4 5 6;7 8 9]; 
w=[5 3 2]; 

y = x > repmat(w,size(x,1),1); 
q = sum(x.*y) ./ sum(y) 

Je ne prétends pas que c'est plus concis.

Answer 3

Voici une façon de résoudre votre problème d'origine: Vous avez un tableau, et vous voulez connaître la moyenne du haut x% de chaque colonne.

%# make up some data 
data = magic(5); 

%# find out how many rows the top 40% are 
nRows = floor(size(data,1)*0.4); 

%# sort the data in descending order 
data = sort(data,1,'descend'); 

%# take the mean of the top 20% of values in each column 
topMean = mean(data(1:nRows,:),1);