从R中的apply()函数返回多个值

时间:2012-09-06 12:52:25

标签: r closures

我想从apply()函数返回多个值并将它们放在R中的单独列中,但我不断收到错误。我想要做的是:

experiments$result1, experiments$result2, experiments$result3 <- apply(experiments, 1, 
function(row)
  #Some analysis here
  #return x, y, and z for column result1, result2, and result3
  x, y, z
)

也许这是解决问题的错误方法。实验是一个包含多列数据的数据框。我想要追加每行的分析结果,但我不知道如何在没有循环的情况下做到这一点,这对于R来说不是惯用的。感谢提前帮助。

所以这里有一些更精确的代码。

experiments$result1, experiments$result2, experiments$result3 <- apply(experiments, 1, function(row)
  x <- row["startingTemp"]*2
  y <- row["startingTemp"]*3
  z <- row["startingTemp"]*4
  return (list(x, y, z))
)

“startingTemp”字段是我的“实验”数据框中的一列。我收到的错误是'闭包'类型不是子集,而且找不到对象'z'。

2 个答案:

答案 0 :(得分:10)

如果要返回的三个值可以放在一个向量中(即它们不是某种复杂类型,如统计测试或拟合模型的结果),只需返回向量,apply将将其绑定到3xN矩阵。

experiments$result <- apply(experiments, 1, function(row){
  x <- row["startingTemp"]*2
  y <- row["startingTemp"]*3
  z <- row["startingTemp"]*4
  c(x, y, z)
})
experiments$result1 <- experiments$result[1,]
experiments$result2 <- experiments$result[2,]
experiments$result3 <- experiments$result[3,]

如果您的三个返回值是复杂类型(或不是标量),请将它们作为列表返回(如Alan建议的那样)并使用lapply / sapply提取它们。

experiment$result1 <- lapply(experiment$result, "[[", 1)

答案 1 :(得分:3)

试试return(list(x=x,y=y,z=z))。对于每个实验,您将返回一个包含3个值的列表。

修改

该函数返回每个实验的ONE列表。

results <- apply(exp,1,function(x) return(list(x=x,y=x*2,z=x*3))) #results is a list of lists`
experiment$res1 <- unlist(results)[attr(unlist(results),"names")=="x"]
experiment$res2 <- unlist(results)[attr(unlist(results),"names")=="y"]
experiment$res3 <- unlist(results)[attr(unlist(results),"names")=="z"]