[問題類型]: 程式諮詢
[軟體熟悉度]: 使用者
[問題敘述]: 想寫一個 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
如果有什麼想法,歡迎請提供線索給我即可,不用全刻出來沒關係。