2012-03-25 26 views
5

que tiene los siguientes datos:Media ponderada por fila

a=c(1:10) 
b=c(16:25) 
c=c(24:33) 
wa=c(3,7,3,3,3,3,3,3,3,1) 
wb=c(3,2,3,3,3,3,3,3,3,8) 
wc=c(4,1,4,4,4,4,4,4,4,1) 
z=data.frame(a,b,c,wa,wb,wc) 

quiero obtener la media ponderada para cada registro. Así que he intentado esto:

weight=apply(subset(z,select=c(wa,wb,wc)),1,function(x) x) 
z$weightMean=apply(subset(z,select=c(a,b,c)),1,function(x) weighted.mean(x,weight)) 

cual devuelve el siguiente mensaje de error:

Error in weighted.mean.default(x, weight) : 
    'x' and 'w' must have the same length 

Así que he intentado esto:

weight=as.vector(weight) 
z$weightMean=apply(subset(z,select=c(a,b,c)),1,function(x) weighted.mean(x,weight)) 

Lo que también devuelve el mismo error.

¿Qué estoy haciendo mal?

Respuesta

3

Esto parece hacer el truco:

> apply(z, 1, function(x) weighted.mean(x[1:3], x[4:6])) 
[1] 14.7 7.3 16.7 17.7 18.7 19.7 20.7 21.7 22.7 24.3 

Esto probablemente será un poco más rápido, aunque menos clara de lo que está pasando:

> rowSums(z[,1:3] * z[,4:6])/rowSums(z[,4:6]) 
[1] 14.7 7.3 16.7 17.7 18.7 19.7 20.7 21.7 22.7 24.3 
+0

Sólo el primer método puede hacer frente a los valores de NA , usando 'weighted.mean (..., na.rm = T)'. El segundo método devuelve NA tan pronto como una de las columnas contenga NA. – Martin

+0

@Martin - 'rowSums()' admite el argumento 'na.rm', por lo que debería funcionar también para ese. – Chase