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

[andrew43 (討厭有好心推文後刪文者)]

[問題類型]: 程式諮詢
[軟體熟悉度]: 使用者
[問題敘述]: 想寫一個 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

如果有什麼想法，歡迎請提供線索給我即可，不用全刻出來沒關係。

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 122.117.37.172
※ 文章網址: https://www.ptt.cc/bbs/R_Language/M.1437668910.A.2A0.html
※ 編輯: andrew43 (122.117.37.172), 07/24/2015 00:29:42
※ 編輯: andrew43 (122.117.37.172), 07/24/2015 00:30:59
※ 編輯: andrew43 (122.117.37.172), 07/24/2015 00:31:54
→ celestialgod: 我給一個想法 明天再寫 頭尾先獨立， 所以剩下只是
→ celestialgod: 組合問題，像是五個元素，n=1的話，中間三個考慮的
→ celestialgod: 是(111), (0), (0)這三個的組合，最後頭尾只要用hea
→ celestialgod: d(, n)還有tail(, n)做新向量插入就好
→ celestialgod: n=2的話 就要頭尾兩個獨立，然後算中間那個
→ celestialgod: 所以要做組合的元素個數是length(v)-n*2
→ celestialgod: 頭尾個數都是n
→ celestialgod: 頭尾都要做到2*n-1個
→ celestialgod: 更正頭尾是從 n到2*n-1
→ andrew43: 你幾乎把過程都寫出來了，謝謝。
推 celestialgod: 我會考慮location matrix(logical) 列數行數都是輸
推 celestialgod: 入向量的長度，預設是FALSE，用迴圈做head, tail做n
推 celestialgod: 個從n到n-1，中間用combinations把，只是要注意comb
推 celestialgod: inations的順序，要運算的值改為TRUE，再用apply做
推 celestialgod: 就好
推 celestialgod: 更正應該是從n到2*n-1
→ andrew43: c兄，多謝。延著你的提示，我突然想到另x<-c(NA,x,NA)
→ andrew43: 之類的辦法就可以解決不少問題。
→ andrew43: 不過這還是有很多限制，例如FUN本身要能處理NA。
→ andrew43: 但對於FUN=sum|mean之類的，這樣就已經足夠了。
→ andrew43: 例如FUN=sum的情況，頭和尾直接填0就可，連NA處理都免了
→ celestialgod: 這是一個方法只是其他計算可能沒用
→ andrew43: 是的。要一般化的話不能這麼做。
推 Edster: 第一個想到的是moving average, 例如hydroTSM::ma
→ Edster: 不過若還要換成別的Function. 應該改成去尾補NA即可.
→ Edster: 原來已經討論過了. 0跟NA不同. 真的建議不要補0
→ andrew43: celestialgod提到，package zoo裡有 roll*() 一系列的
→ andrew43: function，完全是我需要的沒錯。參考看看。
→ Wush978: 有考慮使用filter這個內建函數嗎?
→ andrew43: w兄，我沒搞懂你的意思。