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

作者: celestialgod (天)   2018-01-17 20:43:54
※ 引述《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]
作者: giock18 (小武)   2018-01-18 14:02:00
太強了!這的確就是我所需要的,感謝c大!!!!!關鍵在於前後增列,而且版大讓過成變得很簡潔!

Links booklink

Contact Us: admin [ a t ] ucptt.com