r - 当返回的列数可变时使用 apply 或 plyr

Question

我想知道是否有办法直接从 apply 返回数据帧或 plyr当函数返回的列数可变(但行数始终相同)时调用。例如:

df <- data.frame(A = 1:3, B = c("a","b", "c"))

my_fun <- function(x){
  if(is.numeric(unlist(x))){
    return(x)
  } else {
    return(cbind(x, x))
  }
}

我能够得到的最接近的是返回一个列表并将其转换为数据框:

library(plyr)
data.frame(alply(df, 2, my_fun))
##   A X2.B X2.B.1
## 1 1    a      a
## 2 2    b      b
## 3 3    c      c

感觉应该有一种方法可以在没有额外转换的情况下做到这一点，是吗？

Answer 1

我用 lapply()当您想将函数应用于数据框的多列时，以这种方式很多。在基础 R 中，您可以将数据框视为列表，其中每一列是一个元素。如果你像往常一样使用 lapply() 它将返回一个列表，这不是我们想要的。

> lapply(df, my_fun)
$A
[1] 1 2 3

$B
     x x
[1,] 1 1
[2,] 2 2
[3,] 3 3

但是如果你把结果赋值给 df[]它会向 R 发出信号，表明您想要原始数据框的子集(完整子集，根本不是子集)，从而保留数据框对象类型。

> df[] <- lapply(df, my_fun)
> df
  A B.x B.x
1 1   1   1
2 2   2   2
3 3   3   3