使用mapply通过带有函数的向量元素转换向量元素

时间:2013-08-22 11:27:26

标签: r mapply

有关于mapply的使用的问题。

考虑以下两种情况: 案例1 显示了我希望做的简化示例。我使用mapply来转换我的向量k元素与存储在向量trans中的函数。这有效(related to this question

案例2 中我希望做类似的事情,但是,我想要额外的函数参数(这里,存储在a中)。但我可能想要n个函数参数。我在这个例子中得到的是一个3x3矩阵,在对角线上有预期的结果。我只想要对角线的计算输出。怎么样?

k <- seq(1:3)
# Case 1 -----------------------------------------------------------------------
trans <- c(function(x) x, function(x) 1/x, function(x) x^2)             
# transform vektors elementwise with functions in a "transform" vektor
ktrans   <- mapply(function(f, x) f(x), trans, k)

# 2 -----------------------------------------------------------------------
k <- seq(1:3)
a <- rep(2,3)
transa <- c(function(x,a) x*a, function(x,a) 1/x*a, function(x,a) x^2*a)
ktransa   <- mapply(function(f, x,a) f(x,a), transa, MoreArgs= list(x = k, a= a))

> diag(ktransa)
[1]  2  1 18

1 个答案:

答案 0 :(得分:0)

解决方案:

我没有使用列表将参数发送给函数,而是明确地做了:

# 3 --- Example with n arguments -----------------------------------
k <- seq(1:3)
a <- rep(2,3)
transa <- c(function(x,a) x*a, function(x,a) 1/x*a, function(x,a) x^2*a)
ktransa   <- mapply(function(f, x, a) f(x,a), transa,x = k , a = a)