[问题类型]: 程式咨询
[软件熟悉度]: 使用者
[问题叙述]: 想写一个 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
如果有什么想法,欢迎请提供线索给我即可,不用全刻出来没关系。