[問題] 一維陣列相臨值取平均的實作

作者: andrew43 (討厭有好心推文後刪文者)   2015-07-24 00:28:26
[問題類型]: 程式諮詢
[軟體熟悉度]: 使用者
[問題敘述]: 想寫一個 function 自動求出相鄰元素之平均(或其它統計量)
我想做出一個 funciton,可以做相鄰值的統計量或套用特定的 function。
目前想到的參數有
1. x: 來源 numeric vector
2. n: 取幾個相鄰元素
3. FUN: 想套用的統計量或 function
我的第一個困難是,在頭和在尾的元素在取相鄰元素會有例外。
例如 1:3 的第一個元素是 1,但它沒有上一個元素,
所以就只能往之後的元素納入。
如果是用 for loop,裡頭做例外處理,
我還辦得到,但不知道有沒有更好的寫法。
我的第二個困難是,我想寫成類似 R 中 *apply 系列的 FUN 的風格,
但我不甚了解怎麼撰寫這類風格的 function。
我猜是建出一個 list 再用 lapply 來延伸,不知道好不好?
舉一個例好了
x <- 1:5
newFun(x, n, FUN) <- function{...}
newFun(x, 0, sum) #回傳 1, 2, 3, 4, 5
newFun(x, 1, sum) #回傳 3, 6, 9, 12, 9
# =1+2 =1+2+3 =2+3+4 =3+4+5 =4+5
newFun(x, 2, sum) #回傳 6, 10, 15, 14, 12
newFun(x, 0, function(a){a+1}) #回傳 2,3,4,5,6
如果有什麼想法,歡迎請提供線索給我即可,不用全刻出來沒關係。
作者: celestialgod (天)   2015-07-24 01:50:00
我給一個想法 明天再寫 頭尾先獨立, 所以剩下只是組合問題,像是五個元素,n=1的話,中間三個考慮的是(111), (0), (0)這三個的組合,最後頭尾只要用head(, n)還有tail(, n)做新向量插入就好n=2的話 就要頭尾兩個獨立,然後算中間那個所以要做組合的元素個數是length(v)-n*2頭尾個數都是n頭尾都要做到2*n-1個更正頭尾是從 n到2*n-1
作者: andrew43 (討厭有好心推文後刪文者)   2015-07-24 01:56:00
你幾乎把過程都寫出來了,謝謝。
作者: celestialgod (天)   2015-07-24 01:59:00
我會考慮location matrix(logical) 列數行數都是輸入向量的長度,預設是FALSE,用迴圈做head, tail做n個從n到n-1,中間用combinations把,只是要注意combinations的順序,要運算的值改為TRUE,再用apply做就好更正應該是從n到2*n-1
作者: andrew43 (討厭有好心推文後刪文者)   2015-07-24 02:12:00
c兄,多謝。延著你的提示,我突然想到另x<-c(NA,x,NA)之類的辦法就可以解決不少問題。不過這還是有很多限制,例如FUN本身要能處理NA。但對於FUN=sum|mean之類的,這樣就已經足夠了。例如FUN=sum的情況,頭和尾直接填0就可,連NA處理都免了
作者: celestialgod (天)   2015-07-24 02:18:00
這是一個方法只是其他計算可能沒用
作者: andrew43 (討厭有好心推文後刪文者)   2015-07-24 02:23:00
是的。要一般化的話不能這麼做。
作者: Edster (Edster)   2015-07-24 08:20:00
第一個想到的是moving average, 例如hydroTSM::ma不過若還要換成別的Function. 應該改成去尾補NA即可.原來已經討論過了. 0跟NA不同. 真的建議不要補0
作者: andrew43 (討厭有好心推文後刪文者)   2015-07-24 19:16:00
celestialgod提到,package zoo裡有 roll*() 一系列的function,完全是我需要的沒錯。參考看看。
作者: Wush978 (拒看低質媒體)   2015-07-25 00:54:00
有考慮使用filter這個內建函數嗎?
作者: andrew43 (討厭有好心推文後刪文者)   2015-07-25 12:14:00
w兄,我沒搞懂你的意思。

Links booklink

Contact Us: admin [ a t ] ucptt.com