Vectorizing the loop

This code is similar to the histc function, but each element has weights associated with it. I'm looking to either use an alternate function or to vectorize it to make things faster
I have arrays A and B, and w is an array of weights associated with each element in A.
I'm hoping to calculate:
for i=1:length(B)
x(i) = sum((A>B(i)).*w)
end
Any help is appreciated. Thanks.

5 commentaires

Junaid
Junaid le 24 Avr 2012
I think it can be done, could you show small examples for A, B, w.
Nirmal
Nirmal le 24 Avr 2012
The arrays are fairly long, but to give a really simple example
A = [ 1 3 2 5 3 2 5 6], B = [1 2 3 4 5], w = [0.1 0.03 0.1 0.1 0.05 0.2 0.1 0.01]
Walter Roberson
Walter Roberson le 24 Avr 2012
cumsum(histc(...)) .* w ??
Nirmal
Nirmal le 24 Avr 2012
Not sure if that works - particularly because cumsum(histc(...)) will have the size of B and w will have the size of A.
Sean de Wolski
Sean de Wolski le 24 Avr 2012
+1

Connectez-vous pour commenter.

 Réponse acceptée

Sean de Wolski
Sean de Wolski le 24 Avr 2012

0 votes

x = w*bsxfun(@gt,A.',B)

Plus de réponses (0)

Catégories

En savoir plus sur MATLAB dans Centre d'aide et File Exchange

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by