2014-09-12 14 views
9

Si dispone dei seguenti dati:rollmean con dplyr e magrittr

set.seed(1) 
    data <- data.frame(o=c('a','a','a','a','b','b','b','b','c','c','c','c'), t=c(1,2,3,4,1,2,3,4,1,2,3,4), u=runif(12), v=runif(12)) 
    data 
     o t   u   v 
    1 a 1 0.26550866 0.6870228 
    2 a 2 0.37212390 0.3841037 
    3 a 3 0.57285336 0.7698414 
    4 a 4 0.90820779 0.4976992 
    5 b 1 0.20168193 0.7176185 
    6 b 2 0.89838968 0.9919061 
    7 b 3 0.94467527 0.3800352 
    8 b 4 0.66079779 0.7774452 
    9 c 1 0.62911404 0.9347052 
    10 c 2 0.06178627 0.2121425 
    11 c 3 0.20597457 0.6516738 
    12 c 4 0.17655675 0.1255551 

voglio calcolare la laminazione media (pacchetto zoo) di u per gruppo definito dalla colonnina o. L'ordine per la media mobile è impostato da t. La media mobile deve essere aggiunta come nuova colonna al data.frame.

Voglio usare magrittr e dplyr. Ho provato

data %>% 
     group_by(o) %>% 
     sort(t) %>% 
     select(u) %>% 
     rollmean(3) %>% 
     rbind 

Ma questo non funzionerà. È possibile farlo con magrittr e dplyr o devo farlo passo dopo passo? I valori di o e t sono variabili nei miei dati reali.

Come si riempiono le prime due righe?

risposta

13

Può essere questo aiuta:

library(dplyr) 
data %>% 
group_by(o) %>% 
mutate(rM=rollmean(u,3, na.pad=TRUE, align="right")) 

Se si vuole fare per entrambe le colonne, u e v

fun1 <- function(x) rollmean(x, 3, na.pad=TRUE, align="right") 
data %>% 
group_by(o) %>% 
mutate_each(funs(fun1), u, v) 
+0

fantastico! Se t non è ordinato correttamente, utilizzo i dati%>% group_by (o)%>% arrangement (o, t)%>% mutate (rM = rollmean (u, 3, na.pad = TRUE, align = "right")) – JerryWho

+0

@JerryWho Sì, è possibile utilizzare 'organizzare' quando non è ordinato. – akrun

2

Un involucro più flessibile viene dal pacchetto rowr. Ciò consente di avere finestre di dimensioni diverse all'interno dei dati iniziali.

data %>% 
group_by(o) %>% 
mutate(MEANS = rollApply(u, fun=mean, window=3, align='right'))