Re: [問題] 移動平均的開頭和結尾處理

[celestialgod (天)]

※ 引述《giock18 (小武)》之銘言：
: [問題類型]:
: 程式諮詢(我想用R 做某件事情，但是我不知道要怎麼用R 寫出來)
: [軟體熟悉度]:
: 使用者
: [問題敘述]:
: 遇到一個問題卡關想請教版上大大
: 我有一筆資料，必須依ID分組處理
: 在每組當中，針對每小時的資料，必須往前、往後做移動平均
: 範例:
: Test<-data.frame(hrs=seq(from=0,to=23),TaxiTime=runif(24,min=15,max=25))
: Test$ID<-'a'
: Test2<-Test
: Test2$ID<-'b'
: Test <- rbind(Test,Test2)
: df1 <- group_by(Test,ID) %>%
:   mutate(TaxiTime2 =  lead(TaxiTime, 1), TaxiTime3 = lag(TaxiTime, 1)) %>%
:   filter(!is.na(TaxiTime) | !is.na(TaxiTime2) | !is.na(TaxiTime3))
: %>%data.frame()
: print(df1)
: #在這裡可以看出，這個做法還沒考慮到，遇到00點和23點的時候的問題
: df2 <-mutate(df1, TaxiTime.Mean = rowMeans(select(df1,
: starts_with("TaxiTime")), na.rm = TRUE))
: 但遇到一個問題是，如果是分組的第一列資料(00點的時候)，該如何讓他去取23點的資料
: 去補TaxiTime2 的空值。
: 同理，遇到最後一列資料(23點的時候)該如何讓他去取00點的資料補值。
: 這樣的功能有點像移動平均的頭尾問題，所以我問題這樣取名，
: 如果有更好的問題命名建議我可以修正！！拜託大大了！
: [關鍵字]:
: moving average,分組處理

library(zoo)
library(data.table)
library(pipeR

# 資料生成
DT <- data.table(hr = rep(0:23, 2), id = rep(letters[1:2], each = 24),
                 taxiTime = runif(48, 15, 25))

# pipe版本 (no temporary variables)

out <- DT[hr %in% c(0, 23)] %>>% # 先取出頭尾
  `[`(hr == 23, hr := -1) %>>%   # 讓尾巴變頭
  `[`(hr == 0, hr := 24) %>>%    # 讓頭變尾巴
  rbind(DT) %>>% setorder(id, hr) %>>% # 跟原本資料合併後排序
  `[`(j = mvAvg := rollapply(taxiTime, 3, mean, fill = NA), by = .(id)) %>>%
  # 取moving average
  `[`(hr >= 0 & hr <= 23) # 移除剛剛新增的列

# 或是不用pipe的版本：

tmp <- DT[hr %in% c(0, 23)]
tmp[hr == 23, hr := -1]
tmp[hr == 0, hr := 24]
DT2 <- rbind(tmp, DT)
setorder(DT2, id, hr)
out <- DT2[ , mvAvg := rollapply(taxiTime, 3, mean, fill = NA),
              by = .(id)][hr >= 0 & hr <= 23]


--
R資料整理套件系列文：
magrittr    #1LhSWhpH (R_Language) https://goo.gl/72l1m9
data.table  #1LhW7Tvj (R_Language) https://goo.gl/PZa6Ue
dplyr(上.下)   #1LhpJCfB,#1Lhw8b-s (R_Language) https://goo.gl/I5xX9b
tidyr       #1Liqls1R (R_Language) https://goo.gl/i7yzAz
pipeR       #1NXESRm5 (R_Language) https://goo.gl/zRUISx

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 36.235.47.214
※ 文章網址: https://www.ptt.cc/bbs/R_Language/M.1516193038.A.D3A.html
※ 編輯: celestialgod (36.235.47.214), 01/17/2018 20:57:22
推 giock18: 太強了！這的確就是我所需要的，感謝c大!!!!!
→ giock18: 關鍵在於前後增列，而且版大讓過成變得很簡潔！