Question

假设我有一个包含3个元素的列表 - 每个元素都是一个数据框 - 一个包含3个元素的向量，我想将一个函数应用于列表的每个元素及其在向量中的相应元素。当然我可以用循环来做这个，但我试图用mapply来对它进行矢量化。这就是我正在做的事情

myList = list(data.frame(x = 1:10), data.frame(x = 1:5), data.frame(x = 6:1))
myVec = c(7, 1, 9)
myFun = function(x, y) {
    x$number = y
    return(x)
} 
res = mapply(myFun, myList, myVec)

res不是我的预期：

str(res)
List of 6
 $ : int [1:10] 1 2 3 4 5 6 7 8 9 10
 $ : num [1:10] 7 7 7 7 7 7 7 7 7 7
 $ : int [1:5] 1 2 3 4 5
 $ : num [1:5] 1 1 1 1 1
 $ : int [1:6] 6 5 4 3 2 1
 $ : num [1:6] 9 9 9 9 9 9
 - attr(*, "dim")= int [1:2] 2 3
 - attr(*, "dimnames")=List of 2
  ..$ : chr [1:2] "x" "number"
  ..$ : NULL

我所期望的是一个包含三个元素的列表，每个元素都是一个包含两列的数据框，如下所示：

str(expected)
List of 3
 $ :'data.frame':   10 obs. of  2 variables:
  ..$ x     : int [1:10] 1 2 3 4 5 6 7 8 9 10
  ..$ number: num [1:10] 7 7 7 7 7 7 7 7 7 7
 $ :'data.frame':   5 obs. of  2 variables:
  ..$ x     : int [1:5] 1 2 3 4 5
  ..$ number: num [1:5] 1 1 1 1 1
 $ :'data.frame':   6 obs. of  2 variables:
  ..$ x     : int [1:6] 6 5 4 3 2 1
  ..$ number: num [1:6] 9 9 9 9 9 9

我做错了什么？

Answer 1

我们可以使用Map

Map(myFun, myList, myVec)

由default SIMPLIFY=FALSE使用，即

Map

function (f, ...)  {
f <- match.fun(f)
mapply(FUN = f, ..., SIMPLIFY = FALSE) 
}

Answer 2

试试这个：

res = mapply(myFun, myList, myVec, SIMPLIFY = FALSE)

与列表和矢量mapply

2 个答案: